jax.lax.linalg.cholesky

目录

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