Skip to content

Commit 326a11d

Browse files
ribbon-otterribbon-otterzy1git
authored
ElasticTransform docs: fix issue#8879 (#9350)
Co-authored-by: ribbon-otter <ribbon-otter@github.invalid> Co-authored-by: zy1git <zycoding1@gmail.com>
1 parent 74d1285 commit 326a11d

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

torchvision/transforms/v2/_geometry.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1030,8 +1030,10 @@ class ElasticTransform(Transform):
10301030
see-through-water-like effect.
10311031
10321032
Args:
1033-
alpha (float or sequence of floats, optional): Magnitude of displacements. Default is 50.0.
1034-
sigma (float or sequence of floats, optional): Smoothness of displacements. Default is 5.0.
1033+
alpha (float or sequence of floats, optional): Magnitude of displacements.
1034+
Default is 50.0. A single value is ``[alpha, alpha]``.
1035+
sigma (float or sequence of floats, optional): Smoothness of displacements.
1036+
Default is 5.0. A single value is ``[sigma, sigma]``.
10351037
interpolation (InterpolationMode, optional): Desired interpolation enum defined by
10361038
:class:`torchvision.transforms.InterpolationMode`. Default is ``InterpolationMode.BILINEAR``.
10371039
If input is Tensor, only ``InterpolationMode.NEAREST``, ``InterpolationMode.BILINEAR`` are supported.

0 commit comments

Comments
 (0)