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

[primTorch] Improve narrow and narrow_copy: refs, tests, docs (#87045)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/87045
Approved by: https://github.com/mruberry
This commit is contained in:
Nikita Karetnikov
2022-11-12 00:41:57 +01:00
committed by PyTorch MergeBot
parent 6e5f736d86
commit 4270bb37da
6 changed files with 188 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
torch/_tensor_docs.py
View File

@ -3436,18 +3436,7 @@ add_docstr_all(
r"""
narrow(dimension, start, length) -> Tensor
See :func:`torch.narrow`
Example::
>>> x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> x.narrow(0, 0, 2)
tensor([[ 1, 2, 3],
[ 4, 5, 6]])
>>> x.narrow(1, 1, 2)
tensor([[ 2, 3],
[ 5, 6],
[ 8, 9]])
See :func:`torch.narrow`.
""",
)