Fonction torch.eig PyTorch

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

torch.eigC'est une fonction dans PyTorch utilisée pour calculer les valeurs propres et les vecteurs propres d'une matrice carrée. Elle renvoie toutes les valeurs propres et les vecteurs propres à droite de la matrice.

Définition de fonction

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

Parameters:

  • A(Tensor) : Matrice carrée d'entrée.
  • eigenvectors(bool, facultatif) : Si True, calcule également les vecteurs propres. Par défaut : False.
  • out(tuple, facultatif) : Tuple de sortie.

Valeur de retour:

  • tuple: Renvoie un tuple (valeurs propres, vecteurs propres).

Exemple d'utilisation

Exemple

import torch

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

# Calculer les valeurs propres
eigenvalues, eigenvectors = torch.eig(A, eigenvectors=True)

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

La sortie est :

矩阵 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 参考手册Manuel de référence PyTorch torch

Autres extensions