jax.numpy.linalg.eigvals

jax.numpy.linalg.eigvals(a)

計算一般矩陣的特徵值。

JAX 實作的 numpy.linalg.eigvals()。

參數:

a (ArrayLike) – 形狀為 (..., M, M) 的陣列，用於計算特徵值。

返回:

形狀為 (..., M) 的陣列，包含特徵值。

返回型別:

Array

參見

• jax.numpy.linalg.eig()：計算一般矩陣的特徵值和特徵向量。

• jax.numpy.linalg.eigh()：計算 Hermitian 矩陣的特徵值和特徵向量。

筆記

• 這與 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)