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

compute_uv for SVD (#12517)

Browse Source 
Summary:
Adds a `compute_uv` argument that defaults to `True` for optionally computing the singular vectors during SVD.

Closes https://github.com/pytorch/pytorch/issues/12420 .
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12517

Differential Revision: D10384554

Pulled By: SsnL

fbshipit-source-id: 704998a257afa815eda901b8ae830e8a661695be
This commit is contained in:
vishwakftw
2018-10-15 12:31:15 -07:00
committed by Facebook Github Bot
parent d5eae90537
commit 0740a5d521
16 changed files with 174 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/_tensor_docs.py
View File

@ -2196,7 +2196,7 @@ See :func:`torch.sum`
add_docstr_all('svd',
r"""
svd(some=True) -> (Tensor, Tensor, Tensor)
svd(some=True, compute_uv=True) -> (Tensor, Tensor, Tensor)
See :func:`torch.svd`
""")