jax.numpy.linalg.eigvalsh

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

计算厄米矩阵的特征值。

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

参数:

• a (ArrayLike) – 形状为 (..., M, M) 的数组，包含厄米特矩阵（如果是复数）或对称矩阵（如果是实数）。

• UPLO (str | None) – 指定计算是使用 a 的下三角部分 ('L'，默认) 还是上三角部分 ('U')。

返回:

包含特征值的形状为 (..., M) 的数组，按升序排列。

返回类型:

Array

参见

• jax.numpy.linalg.eig(): 一般特征值分解。

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

示例

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