CODEWITHSUNDEEP
tensorflow
Lenar Hoyt
Lenar Hoyt

Reputation: 6159

How can I implement a convolutional LSTM cell in TensorFlow?

Is the correct general approach to simply copy all of the code of class BasicLSTMCell(RNNCell) and replace all the matrix multiplication with conv2d operations? What are things that I have to keep in mind when implementing it this way?

Upvotes: 1

Views: 2272

Answers (1)

Oliver Hennigh
Oliver Hennigh

Reputation: 56

That is the basic idea. I got an implementation of it working in tensorflow here. It can generate videos that look like this. They seem to work surprisingly well. I edited the rnn_cell.py file to get it working.

Upvotes: 4

Related Questions