A simple way would be to set the batch size to 16*10 and reshape the tensors inside the DataLoader loop into the desired shape.
If you need a specific order of images, you could use e.g. a BatchSampler and apply a sliding window approach inside the __getitem__, but I don’t think this is necessary for your test, since the images are unordered.