cupy.linalg.eigh#

cupy.linalg.eigh(a, UPLO='L')[source]#

返回复数厄米矩阵(共轭对称矩阵)或实对称矩阵的特征值和特征向量。

返回两个对象,一个包含 a 的特征值的 1 维数组,以及一个包含对应特征向量(按列排列)的 2 维方阵或矩阵(取决于输入类型)。

参数
  • a (cupy.ndarray) – 一个对称的 2 维方阵 (M, M) 或一批对称的 2 维方阵 (..., M, M)

  • UPLO (str) – 从 'L''U' 中选择。它指定使用 a 的哪一部分。'L' 使用 a 的下三角部分,'U' 使用 a 的上三角部分。

返回

返回一个元组 (w, v)w 包含特征值,v 包含特征向量。v[:, i] 是对应于特征值 w[i] 的特征向量。对于批量输入,v[k, :, i] 是对应于 a[k] 的特征值 w[k, i] 的特征向量。

返回类型

包含 ndarray 的元组

警告

此函数调用一个或多个 cuSOLVER 例程,如果输入条件不满足,可能会产生无效结果。为了检测这些无效结果,您可以在 cupyx.errstate()cupyx.seterr() 中将 linalg 配置设置为非 ignore 的值。

另请参阅

numpy.linalg.eigh()