原代码链接: ConvLSTM_pytorch 参考资料: LSTM的参数解释 Pytorch-LSTM输入输出参数 ConvLSTM参数详解(Keras) 1.导入pytorch import torch import torch . nn as nn 2.构建ConvLSTMCell class ConvLSTMCell ( nn . Module ) : #这里面全都是数,衡量后面输入数据的维度/通道尺寸 def __init__ ( self , input_dim , hidden_dim , kernel_size , bias ) : super ( ConvLSTMCell , self ) . __init__ ( ) self . input_dim = input_dim self . hidden_dim = hidden_dim #卷积核为一个数组 self . kernel_size = kernel_size #填充为高和宽分别填充的尺寸 self . padding_size = kernel_size [ 0 ] // 2 , kernel_size [ 1 ] // 2 self . bias = bias self . conv = nn . Conv2d ( self . input_dim + self . hidden_dim , 4 *