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,矩阵在对Cholesky分解之前通过计算 \(\frac{1}{2}(x + x^H)\) 进行对称化。如果 False,则只使用 x 的下三角部分;上三角部分被忽略且不访问。 返回: Cholesky 分解作为一个与 x 具有相同 dtype 且形状为 [..., n, n] 的矩阵。如果 Cholesky 分解失败,则返回一个充满 NaN 的矩阵。失败时的行为可能在将来发生变化。 返回类型: Array