You could transform the img multiple times and just return all of them. Inside the DataLoader loop you could then reshape the additional images to the batch dimension and reduce the batch_size of the DataLoader by the same factor.