jax.lax.linalg.cholesky# jax.lax.linalg.cholesky(x, *, symmetrize_input=True)[source]# 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 具有相同的数据类型,形状为 [..., n, n]。如果 Cholesky 分解失败,则返回一个全为 NaN 的矩阵。未来可能会更改失败时的行为。 返回类型: 数组