jax.numpy.linalg.eigvals

jax.numpy.linalg.eigvals(a)

计算一般矩阵的特征值。

JAX 实现 numpy.linalg.eigvals().

参数：

a (ArrayLike) – 形状为 (..., M, M) 的数组，用于计算其特征值。

返回值：

形状为 (..., M) 的数组，包含特征值。

返回类型：

数组

另请参阅

• jax.numpy.linalg.eig()：计算一般矩阵的特征值和特征向量。

• jax.numpy.linalg.eigh()：计算厄米特矩阵的特征值和特征向量。

备注

• 这与numpy.linalg.eigvals()的不同之处在于，jax.numpy.linalg.eigvals()的返回值类型对于32位输入始终为complex64，对于64位输入始终为complex128。

• 目前，非对称特征分解仅在CPU后端实现。

示例

>>> a = jnp.array([[1., 2.],
...                [2., 1.]])
>>> w = jnp.linalg.eigvals(a)
>>> with jnp.printoptions(precision=2):
...  w
Array([ 3.+0.j, -1.+0.j], dtype=complex64)