Функция PyTorch torch.linalg.eig

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.linalg.eigЭто функция в модуле линейной алгебры PyTorch, используемая для вычисления разложения по собственным значениям квадратных матриц. Она возвращает все собственные значения и правые собственные векторы матрицы.

Определение функции

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

Parameters:

  • A(Tensor): Входная квадратная матрица.
  • out(tuple, optional): Выходной кортеж.

Возвращаемое значение:

  • tuple: Возвращает кортеж (собственные значения, собственные векторы).

Пример использования

Пример

import torch

# Создать квадратную матрицу
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)

# Разложение по собственным значениям
eigenvalues, eigenvectors = torch.linalg.eig(A)

print(Матрица A:)
print(A)
print(nСобственные значения:)
print(eigenvalues)
print(nСобственные векторы:)
print(eigenvectors)

Результат вывода:

矩阵 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 参考手册Справочное руководство Pytorch torch

Другие расширения