Torchvision.transfors: How to perform identical transform on both image and target?