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