如何在TensorFlow中实现卷积LSTM单元?

问题描述:

正确的一般方法是简单复制class BasicLSTMCell(RNNCell)的所有代码,并用conv2d操作替换所有的矩阵乘法?在实施这种方式时,我必须记住哪些事情?如何在TensorFlow中实现卷积LSTM单元?

+0

您的意思是“卷积LSTM”?如果是这样,是的,这是基本的想法。您可以使用卷积构建自己的RNNCell的子类。 (可能已经有一个你可以使用,也许别人会知道。) –

这是基本的想法。我得到了一个在tensorflow here工作的实现。它可以生成看起来像this的视频。他们似乎工作得很出色。我编辑了rnn_cell.py文件让它工作。