diff --git a/torchvision/utils.py b/torchvision/utils.py index ab78104d265..7c4f02257e5 100644 --- a/torchvision/utils.py +++ b/torchvision/utils.py @@ -58,7 +58,7 @@ def norm_range(t, range): 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