pytorch
a49406b3 - Fixed batched version of torch.linalg.cond for singular inputs (#58040)

Commit
4 years ago
Fixed batched version of torch.linalg.cond for singular inputs (#58040) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58040 This PR uses `torch.linalg.inv_ex` to determine the non-invertible inputs and return the condition number of infinity for such inputs. Added OpInfo entry for `torch.linalg.cond`. Test Plan: Imported from OSS Reviewed By: ngimel Differential Revision: D28405146 Pulled By: mruberry fbshipit-source-id: 524b9a38309851fa6461cb787ef3fba5aa7d5328
Author
Parents
Loading