Extracting the current state of an RNN layer

General Discussion
, ,
#1

For tf.keras.layers.RNN, there’s a reset_states method that resets the recorded state to some pre-defined input. Is there some way to extract what the current recorded state is?

1 Like
#2

The documentation mentions a states attribute which looks like it’s readable. There’s also a return_state argument in the declaration if you want to include the state in your graph along with your output.

3 Likes
#3

You can also take a look at this example that shows how to retrieve the current state of an RNN layer like LSTM and pass that to initialize another:

3 Likes