mlx.core.linalg.cholesky

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

计算实对称半正定矩阵的 Cholesky 分解。

此函数支持至少为 2 维的数组。当输入维数大于二维时，将对 a 最后两个维度的每个矩阵计算 Cholesky 分解。

如果输入矩阵不是对称半正定的，则行为未定义。

参数:

• a (array) – 输入数组。

• upper (bool, 可选) – 如果为 True，返回上三角 Cholesky 因子。如果为 False，返回下三角 Cholesky 因子。默认值：False。

• stream (Stream, 可选) – 流或设备。默认为 None，此时使用默认设备的默认流。

返回值:

如果 upper = False，返回一个下三角矩阵 L，使得 L @ L.T = a。如果 upper = True，返回一个上三角矩阵 U，使得 U.T @ U = a。

返回类型:

array