jax.numpy.linalg.eigvalsh

jax.numpy.linalg.eigvalsh(a, UPLO='L')

計算 Hermitian 矩陣的特徵值。

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

參數:

• a (ArrayLike) – 形狀為 (..., M, M) 的陣列，包含 Hermitian (如果是複數) 或對稱 (如果是實數) 矩陣。

• UPLO (str | None) – 指定計算是使用 a 的下三角部分 ('L'，預設值) 還是上三角部分 ('U') 完成。

傳回值:

形狀為 (..., M) 的陣列，包含以遞增順序排序的特徵值。

傳回類型:

Array

另請參閱

• jax.numpy.linalg.eig()：一般特徵值分解。

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

範例

>>> a = jnp.array([[1, -2j],
...                [2j, 1]])
>>> w = jnp.linalg.eigvalsh(a)
>>> w
Array([-1.,  3.], dtype=float32)