Tensorflow contrib.rnn中LSTMBlockCell和BasicLSTMCell之间的区别是什么

时间:2017-05-12 03:55:31

标签: tensorflow

我知道LSTMBlockCell在训练开始时初始化是有效的。 Tensorflow的官方API指南说LSTMBlockCell添加了一个forget_bias。我可以在RNN模型中用LSTMBlockCell替换BasicLSTMCell吗? tf.contrib.rnn中有太多东西,我觉得这些API真的不一致。

1 个答案:

答案 0 :(得分:5)

LSTMBlockCellBasicLSTMCell之间的主要区别在于BasicLSTMCell是根据原始TensorFlow操作实现的,因此如果您想要复制和编辑它是有用的代码使用您的LSTM架构。另一方面,LSTMBlockCell使用融合内核,可以获得更好的性能。