Функция torch.eig в PyTorch
Справочное руководство PyTorch torch
torch.eigЭто функция в PyTorch, используемая для вычисления собственных значений и собственных векторов квадратной матрицы. Она возвращает все собственные значения и правые собственные векторы матрицы.
Определение функции
torch.eig(A, eigenvectors=False, out=None)
Parameters:
A(Tensor): Входная квадратная матрица.eigenvectors(bool, optional): Если True, также вычисляет собственные векторы. По умолчанию False.out(tuple, optional): Выходной кортеж.
Возвращаемое значение:
tuple: Возвращает кортеж (собственные значения, собственные векторы).
Пример использования
Пример
import torch
# Создать квадратную матрицу
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
# Вычислить собственные значения
eigenvalues, eigenvectors = torch.eig(A, eigenvectors=True)
print("Матрица A:")
print(A)
print("nСобственные значения:")
print(eigenvalues)
print("nСобственные векторы:")
print(eigenvectors)
# Создать квадратную матрицу
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
# Вычислить собственные значения
eigenvalues, eigenvectors = torch.eig(A, eigenvectors=True)
print("Матрица A:")
print(A)
print("nСобственные значения:")
print(eigenvalues)
print("nСобственные векторы:")
print(eigenvectors)
Вывод:
矩阵 A:
tensor([[1., 2.],
[3., 4.]])
特征值:
tensor([[-0.3723, 0.0000],
[ 5.3723, 0.0000]])
特征向量:
tensor([[-0.8246, -0.4159],
[ 0.5658, -0.9094]])
Другие расширения