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