Libtorch Input Batch

The above is the data preprocessing stage. The batch input to the model is one.
Now, I want to set batch to four, because will process four channels of video at the same time.
How should I do ? Hope get your help. Thank you !