Python Data Augmentation Pytorch Documentation