Fix for min/max returning 0dim tensors now (#469)

1a6038ea · Tongzhou Wang · Francisco Massa · aad3b4ba · 1a6038ea
Commit 1a6038ea authored Apr 12, 2018 by Tongzhou Wang Committed by Francisco Massa Apr 12, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/utils.py torchvision/utils.py +1 -1

No files found.
--- a/torchvision/utils.py
+++ b/torchvision/utils.py
@@ -58,7 +58,7 @@ def make_grid(tensor, nrow=8, padding=2,
            if range is not None:
                norm_ip(t, range[0], range[1])
            else:
-                norm_ip(t, t.min(), t.max())
+                norm_ip(t, t.min().item(), t.max().item())
        if scale_each is True:
            for t in tensor:  # loop over mini-batch dimension