mlx.core.linalg.cholesky

cholesky(a: array, upper: bool = False, *, stream: None | Stream | Device = None) → array

計算實對稱半正定矩陣的 Cholesky 分解。

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.

參數:

• a (array) -- 輸入陣列。

• upper (bool, optional) -- If True, return the upper triangular Cholesky factor. If False, return the lower triangular Cholesky factor. Default: False.

• stream (Stream, optional) -- Stream or device. Defaults to None in which case the default stream of the default device is used.

回傳值:

If upper = False, it returns a lower triangular L matrix such that L @ L.T = a. If upper = True, it returns an upper triangular U matrix such that U.T @ U = a.

回傳值型別:

array