From 7f5b2c42f73dd68712637cd826ee195b108fb2f7 Mon Sep 17 00:00:00 2001 From: dddzg Date: Fri, 18 Sep 2020 21:50:34 +0800 Subject: [PATCH] make the tensor continuous when pass numpy object to tensor (#2483) Co-authored-by: vfdev --- torchvision/transforms/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/transforms/functional.py b/torchvision/transforms/functional.py index 0f884c9edf0..d226c44a429 100644 --- a/torchvision/transforms/functional.py +++ b/torchvision/transforms/functional.py @@ -71,7 +71,7 @@ def to_tensor(pic): if pic.ndim == 2: pic = pic[:, :, None] - img = torch.from_numpy(pic.transpose((2, 0, 1))) + img = torch.from_numpy(pic.transpose((2, 0, 1))).contiguous() # backward compatibility if isinstance(img, torch.ByteTensor): return img.float().div(255)