We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aad3b4b commit 1a6038eCopy full SHA for 1a6038e
torchvision/utils.py
@@ -58,7 +58,7 @@ def norm_range(t, range):
58
if range is not None:
59
norm_ip(t, range[0], range[1])
60
else:
61
- norm_ip(t, t.min(), t.max())
+ norm_ip(t, t.min().item(), t.max().item())
62
63
if scale_each is True:
64
for t in tensor: # loop over mini-batch dimension
0 commit comments