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 1a6038e commit 7bda0e8Copy full SHA for 7bda0e8
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().item(), t.max().item())
+ norm_ip(t, float(t.min()), float(t.max()))
62
63
if scale_each is True:
64
for t in tensor: # loop over mini-batch dimension
0 commit comments