卷积神经网络连续输出多张图片的参数设置？

Question

输入：(30,180)的时序矩阵 用lstm+cnn神经网络模型训练，希望预测5张连续的彩色图片，cnn层应该怎么设置呢？

目前输出1张图片的最后一层cnn：
self.conv1 = nn.Sequential(
nn.Conv2d(in_channels=3, out_channels=3, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)),
nn.ReLU()
)

        cov_1 = self.conv1(tcov_6)
        return cov_1

希望连续输出5张彩色图片，应该怎么设置呢><

Answer 1

那就是batchsize设置为5呗。