Функция PyTorch torch.linalg.eigvals
Справочное руководство Pytorch torch
torch.linalg.eigvalsявляется функцией в модуле линейной алгебры PyTorch, используемой для вычисления собственных значений квадратной матрицы. Она возвращает только собственные значения, а не собственные векторы, и поэтомуtorch.linalg.eigболее эффективна.
Определение функции
torch.linalg.eigvals(A, out=None)
Parameters:
A(Tensor): Входная квадратная матрица.out(Tensor, необязательный): Выходной тензор.
Возвращаемое значение:
torch.Tensor: Возвращает собственные значения.
Пример использования
Пример
import torch
# Создать квадратную матрицу
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)
# Вычислить собственные значения
собственные значения = torch.linalg.eigvals(A)
print("Матрица A:")
print(A)
print("nСобственные значения:")
print(eigenvalues)
# Создать квадратную матрицу
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]], dtype=torch.complex128)
# Вычислить собственные значения
собственные значения = torch.linalg.eigvals(A)
print("Матрица A:")
print(A)
print("nСобственные значения:")
print(eigenvalues)
Вывод:
矩阵 A:
tensor([[1., 2.],
[3., 4.]], dtype=torch.complex128)
特征值:
tensor([-0.3723+0.j, 5.3723+0.j], dtype=torch.complex128)
Другие расширения