Fonction torch.linalg.eig de PyTorch

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

torch.linalg.eigC'est une fonction du module d'algèbre linéaire de PyTorch utilisée pour calculer la décomposition en valeurs propres de matrices carrées. Elle renvoie toutes les valeurs propres et les vecteurs propres à droite de la matrice.

Définition de fonction

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

Parameters:

  • A(Tensor) : Matrice carrée d'entrée.
  • 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]], dtype=torch.complex128)

# Décomposition en valeurs propres
eigenvalues, eigenvectors = torch.linalg.eig(A)

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

Le résultat de sortie est:

矩阵 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)

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

Autres extensions