Keras の RNN/LSTM/GRU で内部状態を取得する - Ahogrammer

自然言語処理で RNN を使っていると、RNN の内部状態を取得したくなることがあります。 TensorFlow では tf.nn.dynamic_rnn 等の関数を使うと、出力と状態を返してくれます。 しかし、Keras でのやり方については意外と日本語の情報がありませんでした。 本記事では Keras で RNN の内部状態を取得する方法についてまとめてみました。 RNN/LSTM/GRU の内部状態を取得 Keras にはリカレント層として、SimpleRNN、LSTM、GRU の3種類が用意されています。これらの層から内部状態を取得するためには、インスタンス化時の引数として return_state=True を渡す必要があります。 return_state を True にすることで、出力に加えて最終状態を取得できるようになります。 では、実際に各リカレント層の内部状態を取得して