我正在尝试重新实现这篇论文《基于语义条件的LSTM自然语言生成用于对话系统》,在这篇论文中,他们在LSTM单元中添加了一个门,并改变了状态的计算方式。
我该如何在TensorFlow中实现这个功能?我是否需要添加一个新的操作符?
回答:
tf.nn.rnn()
和 tf.nn.dynamic_rnn()
函数接受一个类型为 tf.nn.rnn_cell.RNNCell
的参数 cell
。例如,你可以查看 tf.nn.rnn_cell.BasicLSTMCell
的实现(特别是 BasicLSTMCell.__call__()
方法),这可能是你自定义LSTM的一个很好的起点。