to_pil_image changes (#762)

55b5a6b8 · Surgan Jandial · Francisco Massa · 22bc44ed · 55b5a6b8
Commit 55b5a6b8 authored Feb 26, 2019 by Surgan Jandial Committed by Francisco Massa Feb 26, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/transforms/functional.py torchvision/transforms/functional.py +1 -1

No files found.
--- a/torchvision/transforms/functional.py
+++ b/torchvision/transforms/functional.py
@@ -117,7 +117,7 @@ def to_pil_image(pic, mode=None):

        elif pic.ndimension() == 2:
            # if 2D image, add channel dimension (CHW)
-            pic.unsqueeze_(0)
+            pic = pic.unsqueeze(0)

    elif isinstance(pic, np.ndarray):
        if pic.ndim not in {2, 3}: