如何在tensorflow中编写自定义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的良好起点。