Closed
Description
🐛 Bug torchvision.transforms.functional.pil_to_tensor
returns non-contiguous tensor
To Reproduce
from PIL import Image
from torchvision.transforms.functional import pil_to_tensor
pic = Image.open("3_channel_pic.png")
pil_to_tensor(pic).is_contiguous()
# False
Expected behavior
.contiguous()
is recommended to be used after .permute()
, so that
- consistent behavior with
to_tensor()
; - increase computational efficiency
cc @vfdev-5