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

Document torch.quantile interpolation kwarg (#70637)

Browse Source 
Summary:
clone of https://github.com/pytorch/pytorch/pull/59397

This PR documents the interpolation kwarg parameter added in https://github.com/pytorch/pytorch/issues/49267. Now that the forward compatibility period is over, we can expose this parameter.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/70637

Reviewed By: jbschlosser

Differential Revision: D33411707

Pulled By: anjali411

fbshipit-source-id: f5f2d0a6739b3a855bbdf58fc671ac2f0342ce69
This commit is contained in:
Heitor Schueroff
2022-01-05 10:58:23 -08:00
committed by Facebook GitHub Bot
parent 616afcf981
commit 34c49d3d3b
7 changed files with 54 additions and 142 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/_tensor_docs.py
View File

@ -2879,13 +2879,13 @@ Returns the quantization scheme of a given QTensor.
""")
add_docstr_all('quantile', r"""
quantile(q, dim=None, keepdim=False) -> Tensor
quantile(q, dim=None, keepdim=False, *, interpolation='linear') -> Tensor
See :func:`torch.quantile`
""")
add_docstr_all('nanquantile', r"""
nanquantile(q, dim=None, keepdim=False) -> Tensor
nanquantile(q, dim=None, keepdim=False, *, interpolation='linear') -> Tensor
See :func:`torch.nanquantile`
""")