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

Revert "[BE] enable UFMT for torch/nn/functional.py (#128592)"

Browse Source 
This reverts commit f6e6e55fa7d883a89ba99584f8632c260519ba73.

Reverted https://github.com/pytorch/pytorch/pull/128592 on behalf of https://github.com/fbgheith due to breaking internal builds ([comment](https://github.com/pytorch/pytorch/pull/128592#issuecomment-2181783936))
This commit is contained in:
PyTorch MergeBot
2024-06-21 00:44:16 +00:00
parent 9c929f6ce9
commit cc8193c707
18 changed files with 494 additions and 1362 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
torch/_lowrank.py
View File

@ -11,10 +11,7 @@ from torch.overrides import handle_torch_function, has_torch_function
def get_approximate_basis(
A: Tensor,
q: int,
niter: Optional[int] = 2,
M: Optional[Tensor] = None,
A: Tensor, q: int, niter: Optional[int] = 2, M: Optional[Tensor] = None
) -> Tensor:
"""Return tensor :math:`Q` with :math:`q` orthonormal columns such
that :math:`Q Q^H A` approximates :math:`A`. If :math:`M` is
@ -183,10 +180,7 @@ def _svd_lowrank(
def pca_lowrank(
A: Tensor,
q: Optional[int] = None,
center: bool = True,
niter: int = 2,
A: Tensor, q: Optional[int] = None, center: bool = True, niter: int = 2
) -> Tuple[Tensor, Tensor, Tensor]:
r"""Performs linear Principal Component Analysis (PCA) on a low-rank
matrix, batches of such matrices, or sparse matrix.