Python Code For Data Augmentation In Pytorch