我是 TensorFlow 的新手。BaiscLSTMCell
基于 TensorFlow 教程,我设法构建了一个使用 LSTM 训练基本模型的图。
但我需要让它更快。我在这里看到了一个比较,由于我没有 Nvidia GPU,这LSTMBlockFusedCell
似乎是最好的选择。我查看了文档,发现__init__()
和__call__()
函数的签名不同。具体来说,我担心cell_clip
参数 in__init()__
和sequence_length
in 调用。更重要的是,inputs
张量是有形的[time_len, batch_size, input_size]
;这与基本单元格([batch_size, time_len, input_size]
)的不同吗?我不想使用窥视孔,所以我将它留给False
(默认)。
有人可以解释一下和之间是否有任何其他差异(除了性能的改进)BasicLSTMCell
以及LSTMBlockFusedCell
如何正确设置上述参数以达到与原始相同的结果?