我正在尝试重新实现这篇针对口语对话系统的基于语义条件的基于LSTM的自然语言生成的论文,其中他们为LSTM单元添加了一个门,并更改了状态的计算方式。
我如何在张量流中做到这一点?我需要添加一个新的OP吗?
在tf.nn.rnn()
和tf.nn.dynamic_rnn()
函数接受的参数cell
类型tf.nn.rnn_cell.RNNCell
。例如,你可以看看在实施tf.nn.rnn_cell.BasicLSTMCell
(尤其是BasicLSTMCell.__call__()
法),这可能是为您量身定制LSTM一个很好的起点。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句