jax.numpy.linalg.eigvals

jax.numpy.linalg.eigvals(a)

计算一般矩阵的特征值。

numpy.linalg.eigvals() 的 JAX 实现。

参数:

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

返回:

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

返回类型:

Array

另请参阅

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

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

笔记

• 这与 numpy.linalg.eigvals() 的不同之处在于，对于 32 位输入，jax.numpy.linalg.eigvals() 的返回类型始终为 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)