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