Thanks for the information.
It seems as if the output is returned as [seq_len, batch_size, features].
Could you check the shape of the input also?
As you can see, I’m not very familiar with torchtext, but since you are using batch_first=True in yyour nn.LSTM, the batch dimension shoule be in dim0, which is apparently not the case here.