LSTM internal cells for each state

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?