PyTorch torch.eig-Funktion


Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.eigEs ist eine Funktion in PyTorch, die zur Berechnung der Eigenwerte und Eigenvektoren einer quadratischen Matrix verwendet wird. Sie gibt alle Eigenwerte und rechten Eigenvektoren der Matrix zurück.

Funktionsdefinition

torch.eig(A, eigenvectors=False, out=None)

Parameters:

  • A(Tensor): Quadratische Eingabematrix.
  • eigenvectors(bool, optional): Wenn True, werden auch Eigenvektoren berechnet. Standardwert ist False.
  • 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]])

# Eigenwerte berechnen
eigenvalues, eigenvectors = torch.eig(A, eigenvectors=True)

print("Matrix A:")
print(A)
print("nEigenwerte:")
print(eigenvalues)
print("nEigenvektoren:")
print(eigenvectors)

Die Ausgabe ist:

矩阵 A:
tensor([[1., 2.],
        [3., 4.]])
特征值:
tensor([[-0.3723,  0.0000],
        [ 5.3723,  0.0000]])
特征向量:
tensor([[-0.8246, -0.4159],
        [ 0.5658, -0.9094]])

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen