make the tensor continuous when pass numpy object to tensor (#2483)

Co-authored-by: vfdev <vfdev.5@gmail.com>
pytorch · Sep 18, 2020 · 7f5b2c4 · 7f5b2c4
1 parent 5e4a9f6
commit 7f5b2c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)