ptrblck:

transformations rely heavily on PIL functions and thus cannot be applied on Tensor s so easily.


Exactly. If I tranform the tensor to pil and viceversa then the final result is not accurate. How do I solve the problem. I will be highly obliged if you create for mnist dataset. For my test cases, the data loader gives three outputs data, transformation(data) and target.