jax.lax.linalg.cholesky

内容

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 的矩阵。未来可能会更改失败时的行为。

返回类型:

数组