Fonction torch.linalg.eigvals de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.linalg.eigvalsest une fonction du module d'algèbre linéaire de PyTorch utilisée pour calculer les valeurs propres d'une matrice carrée. Elle ne renvoie que les valeurs propres, pas les vecteurs propres, et est donctorch.linalg.eigplus efficace.

Définition de fonction

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

Parameters:

  • A(Tensor) : Matrice carrée d'entrée.
  • out(Tensor, optionnel) : Tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie les valeurs propres.

Exemple d'utilisation

Exemple

import torch

# Créer une matrice carrée
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)

# Calculer les valeurs propres
valeurs propres = torch.linalg.eigvals(A)

print("Matrice A :")
print(A)
print("nValeurs propres :")
print(eigenvalues)

La sortie est :

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions