Hi James,
Thanks for your answer. yes I am using the LSTMCell. The structure is similar to
self.lstm = nn.LSTMCell(in_dim, out_dim)
The vanilla LSTM works fine. Can we use an IF-ELSE statement to control if the LSTMCell gets only the previous hidden state or gets both hidden state and an actual input?
Again, thanks for your help!