machine-learning - 如何在tensorflow中编写定制的LSTM？

Question

我正在尝试重新实现这篇论文Semantically Conditioned LSTM-based Natural Language Generation for Spoken Dialogue Systems ，其中他们向 LSTM 单元添加了一个门并改变了状态的计算方式。

我怎样才能在 tensorflow 中做到这一点？我需要添加新的 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 的一个很好的起点。