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 2b3a5e4 commit 3386083Copy full SHA for 3386083
torchvision/transforms/v2/_misc.py
@@ -17,6 +17,7 @@
17
get_bounding_boxes,
18
get_keypoints,
19
has_any,
20
+ is_cvcuda_tensor,
21
is_pure_tensor,
22
)
23
@@ -240,6 +241,8 @@ class GaussianNoise(Transform):
240
241
Default is True.
242
"""
243
244
+ _transformed_types = Transform._transformed_types + (is_cvcuda_tensor,)
245
+
246
def __init__(self, mean: float = 0.0, sigma: float = 0.1, clip=True) -> None:
247
super().__init__()
248
self.mean = mean
0 commit comments