jax.lax.linalg.schur

jax.lax.linalg.schur(x, *, compute_schur_vectors=True, sort_eig_vals=False, select_callable=None)

Schur 分解。

仅在 CPU 上实现。

计算 Schur 分解

\[A = Q \, U \, Q^{-H}\]

对于方阵 \(A\)。

参数:

• x (ArrayLike) – 一批形状为 [..., m, m] 的方阵。

• compute_schur_vectors (bool) – 如果 True，计算 Schur 向量 :\(Q\)，否则仅计算 \(U\)。

• sort_eig_vals (bool) – 未使用。

• select_callable (Callable[..., Any] | None | None) – 未使用。

返回值:

如果 compute_schur_vectors=True，则返回数组对 U, Q，否则仅返回 U。

返回类型:

tuple[Array, Array]