We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 550656f commit 5fbeac3Copy full SHA for 5fbeac3
torchvision/transforms/v2/_geometry.py
@@ -29,6 +29,7 @@
29
is_pure_tensor,
30
query_size,
31
)
32
+from .functional._utils import is_cvcuda_tensor
33
34
CVCUDA_AVAILABLE = _is_cvcuda_available()
35
@@ -606,6 +607,8 @@ class RandomRotation(Transform):
606
607
608
_v1_transform_cls = _transforms.RandomRotation
609
610
+ _transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
611
+
612
def __init__(
613
self,
614
degrees: Union[numbers.Number, Sequence],
0 commit comments