Passing ouput from model as input pytorch dataloader

Question

I’m doing video prediction frame prediction using LSTM. I have trained the network using input frame at time (t) and target frame at time (t+1).

Now for testing set I