jax.lax.linalg.cholesky

jax.lax.linalg.cholesky(x, *, symmetrize_input=True)

Cholesky 分解。

计算 Cholesky 分解

\[A = L . L^H\]

对于方阵 \(A\)，使得 \(L\) 是下三角矩阵。 \(A\) 的矩阵必须是正定的，如果是复数，则必须是埃尔米特矩阵，如果是实数，则必须是对称矩阵。

参数:

• x (Array) – 一批形状为 [..., n, n] 的埃尔米特（如果为实数则对称）正定方阵。

• symmetrize_input (bool) – 如果 True，则通过计算 \(\frac{1}{2}(x + x^H)\) 在 Cholesky 分解之前将矩阵对称化。 如果 False，则仅使用 x 的下三角； 上三角将被忽略且无法访问。

返回:

Cholesky 分解作为与 x 具有相同 dtype 且形状为 [..., n, n] 的矩阵。 如果 Cholesky 分解失败，则返回一个充满 NaN 的矩阵。 将来可能会更改失败时的行为。

返回类型:

Array