Python Data Augmentation Pytorch Version