Replace with explicit check on image. (#3239)

Summary: Co-authored-by: Francisco Massa <fvsmassa@gmail.com> Reviewed By: datumbox Differential Revision: D26156388 fbshipit-source-id: aa09fdf088ab2271da5ae25d764262d6246e4599
pytorch · Feb 1, 2021 · 9249fb5 · 9249fb5
1 parent 1c2644f
commit 9249fb5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torchvision/transforms/functional_tensor.py b/torchvision/transforms/functional_tensor.py
@@ -59,8 +59,7 @@ def convert_image_dtype(image: torch.Tensor, dtype: torch.dtype = torch.float) -
     if image.dtype == dtype:
         return image
 
-    # TODO: replace with image.dtype.is_floating_point when torchscript supports it
-    if torch.empty(0, dtype=image.dtype).is_floating_point():
+    if image.is_floating_point():
 
         # TODO: replace with dtype.is_floating_point when torchscript supports it
         if torch.tensor(0, dtype=dtype).is_floating_point():