jax.lax.linalg.cholesky

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

Cholesky 分解。

计算 Cholesky 分解

\[A = L . L^H\]

对于方阵 \(A\)，其中 \(L\) 是下三角矩阵。 \(A\) 的矩阵必须是正定的，并且如果为复数，则为 Hermitian 矩阵；如果为实数，则为对称矩阵。

参数:

• x (Array) – 一批形状为 [..., n, n] 的方阵 Hermitian（如果是实数，则为对称）正定矩阵。

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

返回值:

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

返回类型:

Array