ConvLSTM just replace the linear function in normal LSTM with convolution.
I want to use autograd, is this possible to implement simply by using python with no cuda codes or c codes?
ConvLSTM:

image.png749×228 9.23 KB