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 f8aab31 commit 156893fCopy full SHA for 156893f
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
@@ -1045,6 +1046,8 @@ class ElasticTransform(Transform):
1045
1046
1047
_v1_transform_cls = _transforms.ElasticTransform
1048
1049
+ _transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
1050
+
1051
def __init__(
1052
self,
1053
alpha: Union[float, Sequence[float]] = 50.0,
0 commit comments