Wie schreibe ich ein benutzerdefiniertes LSTM in Tensorflow?

Question

Ich versuche, dieses Papier Semantically Conditioned LSTM-based Natural Language Generation for Spoken Dialogue Systems zu re-implementieren, in dem sie ein Gate zu der LSTM-Zelle hinzufügen und ändern, wie der Status berechnet wird.

Wie kann ich dies im Tensorflow tun? Muss ich ein neues OP hinzufügen?

Answer 1

Die Funktionen tf.nn.rnn() und tf.nn.dynamic_rnn() akzeptieren ein Argument cell vom Typ tf.nn.rnn_cell.RNNCell. Zum Beispiel können Sie einen Blick auf die implementation of tf.nn.rnn_cell.BasicLSTMCell (insbesondere die BasicLSTMCell.__call__() method) werfen, die ein guter Ausgangspunkt für Ihre benutzerdefinierte LSTM sein könnte.