How to extract the cell state and hidden state from an RNN model in tensorflow?

Question

I am new to TensorFlow and have difficulties understanding the RNN module. I am trying to extract hidden/cell states from an LSTM. For my code, I am using the implementation fr

Answer 1

One comment about your assumption: the "states" does have only the values of "hidden state" and "memory cell" from last timestep.

The "outputs" contain the "hidden state" from each time step you want (the size of outputs is [batch_size, seq_len, hidden_size]. So I assume that you want "outputs" variable, not "states". See the documentation.