Función PyTorch torch.eig

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.eigEs una función en PyTorch utilizada para calcular los valores propios y vectores propios de una matriz cuadrada. Devuelve todos los valores propios y los vectores propios derechos de la matriz.

Definición de la función

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

Parámetros:

  • A(Tensor): La matriz cuadrada de entrada.
  • eigenvectors(bool, opcional): Si es True, también calcula los vectores propios. Por defecto es False.
  • out(tuple, opcional): Tupla de salida.

Valor de retorno:

  • tuple: Devuelve una tupla (valores propios, vectores propios).

Ejemplo de uso

Ejemplo

import torch

# Crear una matriz cuadrada
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])

# Calcular valores propios
eigenvalues, eigenvectors = torch.eig(A, eigenvectors=True)

print("Matriz A:")
print(A)
print("nValores propios:")
print(eigenvalues)
print("nVectores propios:")
print(eigenvectors)

El resultado de salida es:

矩阵 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 参考手册Manual de referencia de Pytorch torch

Otras extensiones