2016-08-16 74 views
1

正確的一般方法是簡單複製class BasicLSTMCell(RNNCell)的所有代碼,並用conv2d操作替換所有的矩陣乘法?在實施這種方式時,我必須記住哪些事情?如何在TensorFlow中實現卷積LSTM單元?

+0

您的意思是「卷積LSTM」?如果是這樣,是的,這是基本的想法。您可以使用卷積構建自己的RNNCell的子類。 (可能已經有一個你可以使用,也許別人會知道。) –

回答

3

這是基本的想法。我得到了一個在tensorflow here工作的實現。它可以生成看起來像this的視頻。他們似乎工作得很出色。我編輯了rnn_cell.py文件讓它工作。