mlx_rs::linalg

Function cholesky

Source
pub fn cholesky(a: impl AsRef<Array>, upper: Option<bool>) -> Result<Array>
Expand description

Compute the Cholesky decomposition of a real symmetric positive semi-definite matrix.

This function supports arrays with at least 2 dimensions. When the input has more than two dimensions, the Cholesky decomposition is computed for each matrix in the last two dimensions of a.

If the input matrix is not symmetric positive semi-definite, behaviour is undefined.

ยงParams

  • a: input array
  • upper: If true, return the upper triangular Cholesky factor. If false, return the lower triangular Cholesky factor. Default: false.