mlx_rs::linalg

Function cholesky_device

Source
pub fn cholesky_device(
    a: impl AsRef<Array>,
    upper: Option<bool>,
    stream: impl AsRef<Stream>,
) -> 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.