Skip to content

Commit 92f570e

Browse files
committed
Improve docstring
1 parent c8d43c5 commit 92f570e

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

  • src/torchio/transforms/preprocessing/intensity

src/torchio/transforms/preprocessing/intensity/to.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@
1212
class To(IntensityTransform):
1313
"""Convert the image tensor data type and/or device.
1414
15+
This transform is a thin wrapper around :func:`torch.Tensor.to`.
16+
1517
Args:
1618
target: First argument to :func:`torch.Tensor.to`.
1719
to_kwargs: Additional keyword arguments to pass to :func:`torch.Tensor.to`.
@@ -25,7 +27,6 @@ class To(IntensityTransform):
2527
>>> ct_rescaled = rescale(ct_clamped)
2628
>>> to_uint8 = tio.To(torch.uint8)
2729
>>> ct_uint8 = to_uint8(ct_rescaled)
28-
2930
"""
3031

3132
def __init__(

0 commit comments

Comments
 (0)