18 lines
397 B
Python
18 lines
397 B
Python
import torch
|
|
import torch.nn as nn
|
|
|
|
|
|
class Encoder(nn.Module):
|
|
def __init__(self, input_size, hidden_size, output_size):
|
|
super(Encoder, self).__init__()
|
|
|
|
def forward(self, x: torch.Tensor) -> torch.Tensor:
|
|
pass
|
|
|
|
|
|
class Decoder(nn.Module):
|
|
def __init__(self):
|
|
super(Decoder, self).__init__()
|
|
|
|
def forward(self, x: torch.Tensor) -> torch.Tensor:
|
|
pass
|