From 31b5f948b060b03dbef6127027322df44a867cd1 Mon Sep 17 00:00:00 2001 From: SsnL Date: Wed, 11 Apr 2018 18:12:21 -0400 Subject: [PATCH] Fix for min/max returning 0dim tensors now --- torchvision/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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