PyTorch torch.linalg.eigvals 関数
torch.linalg.eigvalsPyTorchの線形代数モジュールで正方行列の固有値を計算する関数です。固有値のみを返し、固有ベクトルを計算しないため、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)
その他の拡張