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 的值。另请参阅