But the ToTensor() I have now in the code is applying it to everything (image, mask). If I make the mask a tensor beforehand, it will apply twice… right? Also what different behaviour would that solution provide?