Función torch.linalg.eig de PyTorch

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

torch.linalg.eigEs una función del módulo de álgebra lineal de PyTorch que se utiliza para calcular la descomposición en valores 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.linalg.eig(A, out=None)

Parámetros:

  • A(Tensor): Matriz cuadrada de entrada.
  • out(tuple, opcional): Tupla de salida.

Valor de retorno:

  • tupleDevuelve una tupla de (valores propios, vectores propios).

Ejemplo de uso

Ejemplo

import torch

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

# Descomposición en valores propios
eigenvalues, eigenvectors = torch.linalg.eig(A)

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.]], 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 参考手册Manual de referencia de Pytorch torch

Otras extensiones