use transformed_types in child classes not transform

Author: justincdavis

torchvision/transforms/v2/_color.py

@@ -7,7 +7,7 @@
 from torchvision.transforms.v2 import functional as F, Transform
 
 from ._transform import _RandomApplyTransform
-from ._utils import query_chw
+from ._utils import is_cvcuda_tensor, query_chw
 
 
 class Grayscale(Transform):
@@ -22,6 +22,8 @@ class Grayscale(Transform):
 
     _v1_transform_cls = _transforms.Grayscale
 
+    _transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
+
     def __init__(self, num_output_channels: int = 1):
         super().__init__()
         self.num_output_channels = num_output_channels
@@ -44,6 +46,8 @@ class RandomGrayscale(_RandomApplyTransform):
 
     _v1_transform_cls = _transforms.RandomGrayscale
 
+    _transformed_types = _RandomApplyTransform._transformed_types + (is_cvcuda_tensor,)
+
     def __init__(self, p: float = 0.1) -> None:
         super().__init__(p=p)
 
@@ -62,6 +66,8 @@ class RGB(Transform):
     to have [..., 1 or 3, H, W] shape, where ... means an arbitrary number of leading dimensions
     """
 
+    _transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
+
     def __init__(self):
         super().__init__()