PyTorch torch.linalg.eigvals Funktion


Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.linalg.eigvalsist eine Funktion im Modul für lineare Algebra von PyTorch, die verwendet wird, um Eigenwerte einer quadratischen Matrix zu berechnen. Sie gibt nur Eigenwerte zurück, keine Eigenvektoren, und ist dahertorch.linalg.eigeffizienter.

Funktionsdefinition

torch.linalg.eigvals(A, out=None)

Parameters:

  • A(Tensor): Quadratische Eingabematrix.
  • out(Tensor, optional): Ausgabe-Tensor.

Rückgabewert:

  • torch.Tensor: Gibt Eigenwerte zurück.

Anwendungsbeispiel

Beispiel

import torch

# Quadratische Matrix erstellen
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)

# Eigenwerte berechnen
Eigenwerte = torch.linalg.eigvals(A)

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

Die Ausgabe ist:

矩阵 A:
tensor([[1., 2.],
        [3., 4.]], dtype=torch.complex128)
特征值:
tensor([-0.3723+0.j,  5.3723+0.j], dtype=torch.complex128)

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen