将模型的输出作为输入pytorch数据加载器传递

Question

我正在使用LSTM进行视频预测帧预测。我已经使用时间（t）的输入帧和时间（t + 1）的目标帧训练了网络。

现在用于测试集了，我希望LSTM将时间（t）的输出帧用作时间（t + 1）的输入帧，是否可以使用数据加载器来做到这一点？