Yes, your approach would technically work (just remove the forward call and use model(i) instead).
I don’t know what model you are using, but in case it doesn’t use “states” from previous iterations (e.g. via RNNs), it would not learn any temporal dependency and would treat each input as a new sample.