The pytorch LSTM returns the cell state for the last time step only. How do I get the cell state or the forget gate for each timestep? I am using a single layer LSTM.
3 Likes
I have the same question. It would be great if this could be provided so we don’t have to manually compute LSTMCells for each timestep
Hi, recently, I have the same problem. Do you get a better solution?