You can always apply the same transformation if you specify the parameter and use the functional versions of transformations: torchvision.transforms.functional — Torchvision 0.10.0 documentation (pytorch.org) 1.


ToTensor should already normalize to [0, 1]. torchvision.transforms.transforms — Torchvision 0.10.0 documentation (pytorch.org)