PyTorch torch.linalg.eigvals 函数
Pytorch torch 参考手册torch.linalg.eigvals 是 PyTorch 线性代数模块中用于计算方阵特征值的函数。它只返回特征值，不计算特征向量，因此比 torch.linalg.eig 更高效。
函数定义
torch.linalg.eigvals(A, out=None)
参数:
A(Tensor): 输入方阵。out(Tensor, 可选): 输出张量。
返回值:
torch.Tensor: 返回特征值。
使用示例
实例
import torch
# 创建方阵
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)
# 计算特征值
eigenvalues = torch.linalg.eigvals(A)
print("矩阵 A:")
print(A)
print("n特征值:")
print(eigenvalues)
# 创建方阵
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)
# 计算特征值
eigenvalues = torch.linalg.eigvals(A)
print("矩阵 A:")
print(A)
print("n特征值:")
print(eigenvalues)
输出结果为：
矩阵 A:
tensor([[1., 2.],
[3., 4.]], dtype=torch.complex128)
特征值:
tensor([-0.3723+0.j, 5.3723+0.j], dtype=torch.complex128)