mlx.core.linalg.cholesky_inv

mlx.core.linalg.cholesky_inv#

cholesky_inv(L: array, upper: bool = False, *, stream: None | Stream | Device = None) → array#

使用乔利斯基分解计算实对称半正定矩阵的逆。

设 \(\mathbf{A}\) 为实对称半正定矩阵，\(\mathbf{L}\) 为其乔利斯基分解，满足

\[\begin{aligned} \mathbf{A} = \mathbf{L}\mathbf{L}^T \end{aligned}\]

此函数计算 \(\mathbf{A}^{-1}\)。

此函数支持至少包含 2 个维度的数组。当输入具有两个以上维度时，将对 \(\mathbf{L}\) 的最后两个维度中的每个矩阵计算乔利斯基逆。

如果输入矩阵不是三角矩阵，则行为未定义。

参数:

  • L (array) – 输入数组。

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

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

返回值:

\(\mathbf{A^{-1}}\)，其中 \(\mathbf{A} = \mathbf{L}\mathbf{L}^T\)。

返回类型:

array

目录