PyTorch torch.linalg.eigvals función

Pytorch torch 参考手册Pytorch torch manual de referencia

torch.linalg.eigvalses la función del módulo de álgebra lineal de PyTorch para calcular valores propios de una matriz cuadrada. Solo devuelve los valores propios, no calcula los vectores propios, por lo que es mástorch.linalg.eigeficiente.

Definición de la función

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

Parámetros:

  • A(Tensor): la matriz cuadrada de entrada.
  • out(Tensor, opcional): tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve los valores propios.

Ejemplo de uso

Ejemplo

import torch

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

# Calcular valores propios
eigenvalues = torch.linalg.eigvals(A)

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

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)

Pytorch torch 参考手册Pytorch torch manual de referencia

Otras extensiones