Add minor comment about ToTensor (#8300)

pytorch · Mar 13, 2024 · dd66a9d · dd66a9d
1 parent bdd690e
commit dd66a9d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchvision/transforms/v2/_deprecated.py b/torchvision/transforms/v2/_deprecated.py
@@ -17,6 +17,7 @@ class ToTensor(Transform):
     .. warning::
         :class:`v2.ToTensor` is deprecated and will be removed in a future release.
         Please use instead ``v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])``.
+        Output is equivalent up to float precision.
 
     This transform does not support torchscript.
 
@@ -41,6 +42,7 @@ def __init__(self) -> None:
         warnings.warn(
             "The transform `ToTensor()` is deprecated and will be removed in a future release. "
             "Instead, please use `v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])`."
+            "Output is equivalent up to float precision."
         )
         super().__init__()