jax.lax.linalg.cholesky

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

Cholesky 分解。

计算 Cholesky 分解

\[A = L . L^H\]

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

参数:

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

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

返回:

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

返回类型:

数组