1

我是 TensorFlow 的新手。BaiscLSTMCell基于 TensorFlow 教程,我设法构建了一个使用 LSTM 训练基本模型的图。

但我需要让它更快。我在这里看到了一个比较,由于我没有 Nvidia GPU,这LSTMBlockFusedCell似乎是最好的选择。我查看了文档,发现__init__()__call__()函数的签名不同。具体来说,我担心cell_clip参数 in__init()__sequence_lengthin 调用。更重要的是,inputs张量是有形的[time_len, batch_size, input_size];这与基本单元格([batch_size, time_len, input_size])的不同吗?我不想使用窥视孔,所以我将它留给False(默认)。

有人可以解释一下和之间是否有任何其他差异(除了性能的改进)BasicLSTMCell以及LSTMBlockFusedCell如何正确设置上述参数以达到与原始相同的结果?

4

1 回答 1

0

的文档LSTMBlockCell说它应该与 兼容LSTMCell,因此相同的参数应该具有相同的含义。

输入张量是批处理优先还是时间优先与单元无关,而是与您正在使用的 dynamic_rnn / static_rnn 相关。

于 2017-09-05T15:52:48.257 回答