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

Add missing parameters to tensor docs (#2541)

Browse Source
This commit is contained in:
Sam Gross
2017-08-25 17:40:55 -04:00
committed by Soumith Chintala
parent 50129befb6
commit 327a0793b4
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
torch/_tensor_docs.py
View File

@ -725,7 +725,7 @@ Torch C API as the given tensor.
add_docstr_all('kthvalue',
"""
kthvalue(k, dim=None) -> (Tensor, LongTensor)
kthvalue(k, dim=None, keepdim=False) -> (Tensor, LongTensor)
See :func:`torch.kthvalue`
""")
@ -867,28 +867,28 @@ See :func:`torch.masked_select`
add_docstr_all('max',
"""
max(dim=None) -> float or (Tensor, Tensor)
max(dim=None, keepdim=False) -> float or (Tensor, Tensor)
See :func:`torch.max`
""")
add_docstr_all('mean',
"""
mean(dim=None) -> float or (Tensor, Tensor)
mean(dim=None, keepdim=False) -> float or (Tensor, Tensor)
See :func:`torch.mean`
""")
add_docstr_all('median',
"""
median(dim=-1, values=None, indices=None) -> (Tensor, LongTensor)
median(dim=None, keepdim=False) -> (Tensor, LongTensor)
See :func:`torch.median`
""")
add_docstr_all('min',
"""
min(dim=None) -> float or (Tensor, Tensor)
min(dim=None, keepdim=False) -> float or (Tensor, Tensor)
See :func:`torch.min`
""")
@ -902,7 +902,7 @@ See :func:`torch.mm`
add_docstr_all('mode',
"""
mode(dim=-1, values=None, indices=None) -> (Tensor, LongTensor)
mode(dim=None, keepdim=False) -> (Tensor, LongTensor)
See :func:`torch.mode`
""")
@ -1012,7 +1012,7 @@ See :func:`torch.nonzero`
add_docstr_all('norm',
"""
norm(p=2) -> float
norm(p=2, dim=None, keepdim=False) -> float
See :func:`torch.norm`
""")
@ -1092,7 +1092,7 @@ In-place version of :meth:`~Tensor.pow`
add_docstr_all('prod',
"""
prod() -> float
prod(dim=None, keepdim=False) -> float
See :func:`torch.prod`
""")
@ -1408,7 +1408,7 @@ In-place version of :meth:`~Tensor.squeeze`
add_docstr_all('std',
"""
std() -> float
std(dim=None, unbiased=True, keepdim=False) -> float
See :func:`torch.std`
""")
@ -1465,7 +1465,7 @@ In-place version of :meth:`~Tensor.sub`
add_docstr_all('sum',
"""
sum(dim=None) -> float
sum(dim=None, keepdim=False) -> float
See :func:`torch.sum`
""")
@ -1682,7 +1682,7 @@ In-place version of :meth:`~Tensor.unsqueeze`
add_docstr_all('var',
"""
var() -> float
var(dim=None, unbiased=True, keepdim=False) -> float
See :func:`torch.var`
""")