Skip to content

Commit 3af1d6b

Browse files
fix markdown
1 parent 1d05f58 commit 3af1d6b

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

torchvision/transforms/v2/_type_conversion.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def transform(self, inpt: Any, params: dict[str, Any]) -> torch.Tensor:
9696

9797

9898
class ToCVCUDATensor(Transform):
99-
"""Convert a `torch.Tensor` with NCHW layout to a `cvcuda.Tensor` with NHWC layout.
99+
"""Convert a ``torch.Tensor`` with NCHW layout to a ``cvcuda.Tensor`` with NHWC layout.
100100
If the input tensor is on CPU, it will automatically be transferred to GPU.
101101
Only 1-channel and 3-channel images are supported.
102102
@@ -116,7 +116,7 @@ def transform(self, inpt: torch.Tensor, params: dict[str, Any]) -> "cvcuda.Tenso
116116

117117

118118
class CVCUDAToTensor(Transform):
119-
"""Convert a `cvcuda.Tensor` with NHWC layout to a `torch.Tensor` with NCHW layout.
119+
"""Convert a ``cvcuda.Tensor`` with NHWC layout to a ``torch.Tensor`` with NCHW layout.
120120
121121
This function does not support torchscript.
122122

torchvision/transforms/v2/functional/_type_conversion.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,12 @@ def to_image(inpt: Union[torch.Tensor, PIL.Image.Image, np.ndarray]) -> tv_tenso
3535

3636
@torch.jit.unused
3737
def to_cvcuda_tensor(inpt: torch.Tensor) -> "cvcuda.Tensor":
38-
"""See :class:`~torchvision.transforms.v2.ToCVCUDATensor` for details."""
38+
"""See :class:``~torchvision.transforms.v2.ToCVCUDATensor`` for details."""
3939
cvcuda = _import_cvcuda()
4040
if not torch.jit.is_scripting() and not torch.jit.is_tracing():
4141
_log_api_usage_once(to_cvcuda_tensor)
4242
if not isinstance(inpt, torch.Tensor):
43-
raise TypeError(f"inpt should be `torch.Tensor`. Got {type(inpt)}.")
43+
raise TypeError(f"inpt should be ``torch.Tensor``. Got {type(inpt)}.")
4444
if inpt.ndim != 4:
4545
raise ValueError(f"pic should be 4 dimensional. Got {inpt.ndim} dimensions.")
4646
# Convert to NHWC as CVCUDA transforms do not support NCHW
@@ -50,12 +50,12 @@ def to_cvcuda_tensor(inpt: torch.Tensor) -> "cvcuda.Tensor":
5050

5151
@torch.jit.unused
5252
def cvcuda_to_tensor(cvcuda_img: "cvcuda.Tensor") -> torch.Tensor:
53-
"""See :class:`~torchvision.transforms.v2.CVCUDAToTensor` for details."""
53+
"""See :class:``~torchvision.transforms.v2.CVCUDAToTensor`` for details."""
5454
cvcuda = _import_cvcuda()
5555
if not torch.jit.is_scripting() and not torch.jit.is_tracing():
5656
_log_api_usage_once(cvcuda_to_tensor)
5757
if not isinstance(cvcuda_img, cvcuda.Tensor):
58-
raise TypeError(f"cvcuda_img should be `cvcuda.Tensor`. Got {type(cvcuda_img)}.")
58+
raise TypeError(f"cvcuda_img should be ``cvcuda.Tensor``. Got {type(cvcuda_img)}.")
5959
cuda_tensor = torch.as_tensor(cvcuda_img.cuda(), device="cuda")
6060
if cvcuda_img.ndim != 4:
6161
raise ValueError(f"Image should be 4 dimensional. Got {cuda_tensor.ndim} dimensions.")

0 commit comments

Comments
 (0)