frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-22 14:15:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move argsort to C++

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17099

Differential Revision: D14165671

Pulled By: ezyang

fbshipit-source-id: 3871de6874fe09871ebd9b8943c13c9af325bf33
This commit is contained in:
Gao, Xiang
2019-02-21 07:50:27 -08:00
committed by Facebook Github Bot
parent 37890610b0
commit 722cbe3064
10 changed files with 55 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
torch/_tensor_docs.py
View File

@ -2206,11 +2206,18 @@ Example::
add_docstr_all('sort',
r"""
sort(dim=None, descending=False) -> (Tensor, LongTensor)
sort(dim=-1, descending=False) -> (Tensor, LongTensor)
See :func:`torch.sort`
""")
add_docstr_all('argsort',
r"""
argsort(dim=-1, descending=False) -> LongTensor
See :func: `torch.argsort`
""")
add_docstr_all('sparse_dim',
r"""
sparse_dim() -> int