torch.linalg.cholesky_ex
torch.linalg.cholesky_ex(A, *, upper=False, check_errors=False, out=None)
Computes the Cholesky decomposition of a complex Hermitian or real
   symmetric positive-definite matrix.
This function skips the (slow) error checking and error message
   construction of "torch.linalg.cholesky()", instead directly
   returning the LAPACK error codes as part of a named tuple "(L,
   info)". This makes this function a faster way to check if a matrix
   is positive-definite, and it provides an opportunity to handle
   decomposition errors more gracefully or performantly than
   "torch.linalg.cholesky()" does.
Supports input of float, double, cfloat and cdouble dtypes. Also
   supports batches of matrices, and if "A" is a batch of matrices
   then the output has the same batch dimensions.
If "A" is not a Hermitian positive-definite matrix, or if it's a
   batch of matrices and one or more of them is not a Hermitian