Solved by cskarthik7 in post #2 
              
              
                import torch
from torchvision import transforms
transform = transforms.Compose([
  transforms.Resize((400,400)),
  transforms.RandomHorizontalFlip(p=0.5),
  transforms.RandomVerticalFlip(p=0.5),
  transforms.RandomRotation(60)
])
z = transforms.Resize((400,400))
x = torch.ones(64, 2048, 2048)
y = tr…