jax.lax.linalg.eig

jax.lax.linalg.eig(x, *, compute_left_eigenvectors=True, compute_right_eigenvectors=True)

一般矩阵的特征分解。

目前非对称特征分解仅在 CPU 上实现。

参数：:

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

• compute_left_eigenvectors (bool) – 如果为真，将计算左特征向量。

• compute_right_eigenvectors (bool) – 如果为真，将计算右特征向量。

返回：:

矩阵 x 的特征分解，是一个元组，形式为 (w, vl, vr)，其中 w 是特征值，vl 是左特征向量，vr 是右特征向量。 vl 和 vr 是可选的，只有当 compute_left_eigenvectors 或 compute_right_eigenvectors 分别为 True 时才会包含。

如果特征分解失败，则该批次元素将返回全为 NaN 的数组。

返回值类型:

list[Array]