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]