Pytorch augmentation transforms.