Transforms In Tensorflow Vs Pytorch