Python Data Augmentation Pytorch Vs Tensorflow