I need to load a batch of 16 images and process it (Step 1), and then I need to apply a different transformation to each of these images and then process it (Step 2). I need to iterate Step 2 few times. How to do it in PyTorch?