@Flo Why we need self.encoder = nn.Embedding(input_size, embed_size) suppose I have 3D tensor [64, 10, 128] where 64 is batch size, 10 is sequence length, and 128 is feature size. What should I do with inp ?