pytorch
6ba9fd59 - Added "Tensor tol" overload of torch.linalg.matrix_rank (#54157)

Commit
4 years ago
Added "Tensor tol" overload of torch.linalg.matrix_rank (#54157) Summary: Currently `torch.linalg.matrix_rank` accepts only Python's float for `tol=` argument. The current behavior is not NumPy compatible and this PR adds the possibility to pass Tensor for matrix-wise tolerances. Ref. https://github.com/pytorch/pytorch/issues/42666 Pull Request resolved: https://github.com/pytorch/pytorch/pull/54157 Reviewed By: ezyang Differential Revision: D27961548 Pulled By: mruberry fbshipit-source-id: 47318eefa07a7876e6360dae089e5389b9939489
Author
Parents
Loading