I have a Sequential model in PyTorch:
model = nn.Sequential( nn.Embedding(alphabet_size, 64), nn.LSTM(64, stateful=False, return_sequences=Tru