Skip to content

Commit 3386083

Browse files
committed
add cvcuda tensor to transformed types of GaussianNoise
1 parent 2b3a5e4 commit 3386083

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchvision/transforms/v2/_misc.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
get_bounding_boxes,
1818
get_keypoints,
1919
has_any,
20+
is_cvcuda_tensor,
2021
is_pure_tensor,
2122
)
2223

@@ -240,6 +241,8 @@ class GaussianNoise(Transform):
240241
Default is True.
241242
"""
242243

244+
_transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
245+
243246
def __init__(self, mean: float = 0.0, sigma: float = 0.1, clip=True) -> None:
244247
super().__init__()
245248
self.mean = mean

0 commit comments

Comments
 (0)