frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix documentation for tensor.repeat. (#131195)

Browse Source 
Fixes #130930.

Adjusts the documentation which used `sizes` instead of `repeats`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131195
Approved by: https://github.com/mikaylagawarecki, https://github.com/soulitzer
This commit is contained in:
Miguel Perez
2024-07-22 14:48:18 +00:00
committed by PyTorch MergeBot
parent 26383a6cc0
commit 5c78581fc9
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torch/_tensor_docs.py
View File

@ -4159,7 +4159,7 @@ In-place version of :meth:`~Tensor.renorm`
add_docstr_all( add_docstr_all(
"repeat", "repeat",
r""" r"""
repeat(*sizes) -> Tensor repeat(*repeats) -> Tensor
Repeats this tensor along the specified dimensions. Repeats this tensor along the specified dimensions.
@ -4174,8 +4174,7 @@ Unlike :meth:`~Tensor.expand`, this function copies the tensor's data.
For the operator similar to `numpy.repeat`, see :func:`torch.repeat_interleave`. For the operator similar to `numpy.repeat`, see :func:`torch.repeat_interleave`.
Args: Args:
sizes (torch.Size or int...): The number of times to repeat this tensor along each repeat (torch.Size, int..., tuple of int or list of int): The number of times to repeat this tensor along each dimension
dimension
Example:: Example::