jax.lax.linalg.cholesky

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

Cholesky 分解。

計算 Cholesky 分解

\[A = L . L^H\]

針對方陣 \(A\)，使得 \(L\) 為下三角矩陣。\(A\) 的矩陣必須是正定且為 Hermitian（若為複數）或對稱（若為實數）。

參數:

• x (Array) – 一批形狀為 [..., n, n] 的方陣 Hermitian（若為實數則為對稱）正定矩陣。

• symmetrize_input (bool) – 若為 True，則在 Cholesky 分解之前，透過計算 \(\frac{1}{2}(x + x^H)\) 將矩陣對稱化。若為 False，則僅使用 x 的下三角部分；上三角部分會被忽略且不會被存取。

傳回:

Cholesky 分解，以與 x 相同 dtype 和形狀 [..., n, n] 的矩陣形式傳回。若 Cholesky 分解失敗，則傳回一個充滿 NaNs 的矩陣。失敗時的行為未來可能會變更。

傳回類型:

Array