Torchvision Transforms Functional Tensor