cunumeric.linalg.cholesky#

cunumeric.linalg.cholesky(a: ndarray) ndarray#

Cholesky decomposition.

Return the Cholesky decomposition, L * L.H, of the square matrix a, where L is lower-triangular and .H is the conjugate transpose operator (which is the ordinary transpose if a is real-valued). a must be Hermitian (symmetric if real-valued) and positive-definite. No checking is performed to verify whether a is Hermitian or not. In addition, only the lower-triangular and diagonal elements of a are used. Only L is actually returned.

Parameters:

a ((..., M, M) array_like) – Hermitian (symmetric if all elements are real), positive-definite input matrix.

Returns:

L – Upper or lower-triangular Cholesky factor of a. Returns a matrix object if a is a matrix object.

Return type:

(…, M, M) array_like

Notes

The current implementation kills the process when the decomposition fails.

Availability:

Multiple GPUs, Multiple CPUs