テンソルフローでカスタマイズされたLSTMを書くには？

Question

このペーパーSemantically Conditioned LSTM-based Natural Language Generation for Spoken Dialogue Systemsを再実装しようとしていますが、LSTMセルにゲートを追加し、状態の計算方法を変更します。

テンソルフローでこれを行うにはどうすればよいですか？新しいOPを追加する必要はありますか？

Answer 1

tf.nn.rnn()およびtf.nn.dynamic_rnn()関数は、tf.nn.rnn_cell.RNNCellの引数cellを受け入れます。たとえば、implementation of tf.nn.rnn_cell.BasicLSTMCell（特にBasicLSTMCell.__call__() method）を見ることができます。これは、カスタマイズしたLSTMの出発点になる場合があります。