Skip to content

Commit 156893f

Browse files
committed
update transformed types
1 parent f8aab31 commit 156893f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchvision/transforms/v2/_geometry.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
is_pure_tensor,
3030
query_size,
3131
)
32+
from .functional._utils import is_cvcuda_tensor
3233

3334
CVCUDA_AVAILABLE = _is_cvcuda_available()
3435

@@ -1045,6 +1046,8 @@ class ElasticTransform(Transform):
10451046

10461047
_v1_transform_cls = _transforms.ElasticTransform
10471048

1049+
_transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
1050+
10481051
def __init__(
10491052
self,
10501053
alpha: Union[float, Sequence[float]] = 50.0,

0 commit comments

Comments
 (0)