Pytorch Data Augmentations Definition