`torchvision` v2 transforms not applying to both image and mask

The functional API works.