@Dhorka

However, I didn’t see that “transforms” functionality in the pytorch documentation.

You are right about the lack of documentation about this. Have a look at this issue 11. The author had a similar question.