PyTorch torch.linalg.eig-Funktion
PyTorch torch Referenzhandbuch
torch.linalg.eigEs ist eine Funktion im PyTorch-Modul für lineare Algebra, die verwendet wird, um die Eigenwertzerlegung quadratischer Matrizen zu berechnen. Sie gibt alle Eigenwerte und rechten Eigenvektoren der Matrix zurück.
Funktionsdefinition
torch.linalg.eig(A, out=None)
Parameters:
A(Tensor): Quadratische Eingabematrix.out(tuple, optional): Ausgabetupel.
Rückgabewert:
tuple: Gibt ein Tupel aus (Eigenwerte, Eigenvektoren) zurück.
Anwendungsbeispiel
Beispiel
import torch
# Quadratische Matrix erstellen
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)
# Eigenwertzerlegung
eigenvalues, eigenvectors = torch.linalg.eig(A)
print(Matrix A:)
print(A)
print(nEigenwerte:)
print(eigenvalues)
print(nEigenvektoren:)
print(eigenvectors)
# Quadratische Matrix erstellen
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)
# Eigenwertzerlegung
eigenvalues, eigenvectors = torch.linalg.eig(A)
print(Matrix A:)
print(A)
print(nEigenwerte:)
print(eigenvalues)
print(nEigenvektoren:)
print(eigenvectors)
Das Ausgabeergebnis ist:
矩阵 A:
tensor([[1., 2.],
[3., 4.]], dtype=torch.complex128)
特征值:
tensor([-0.3723+0.j, 5.3723+0.j], dtype=torch.complex128)
特征向量:
tensor([[-0.8246+0.j, -0.4159+0.j],
[ 0.5658+0.j, -0.9094+0.j]], dtype=torch.complex128)
Andere Erweiterungen