PyTorch torch.eig Function
PyTorch torch Reference Manual
torch.eigIt is a function in PyTorch used to compute eigenvalues and eigenvectors of a square matrix. It returns all eigenvalues and right eigenvectors of the matrix.
Function Definition
torch.eig(A, eigenvectors=False, out=None)
Parameters:
A(Tensor): Input square matrix.eigenvectors(bool, optional): If True, also computes eigenvectors. Defaults to False.out(tuple, optional): Output tuple.
Return Value:
tuple: Returns a tuple of (eigenvalues, eigenvectors).
Usage Example
Example
import torch
# Create a square matrix
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
# Compute eigenvalues
eigenvalues, eigenvectors = torch.eig(A, eigenvectors=True)
print("Matrix A:")
print(A)
print("nEigenvalues:")
print(eigenvalues)
print("nEigenvectors:")
print(eigenvectors)
# Create a square matrix
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
# Compute eigenvalues
eigenvalues, eigenvectors = torch.eig(A, eigenvectors=True)
print("Matrix A:")
print(A)
print("nEigenvalues:")
print(eigenvalues)
print("nEigenvectors:")
print(eigenvectors)
The output is:
矩阵 A:
tensor([[1., 2.],
[3., 4.]])
特征值:
tensor([[-0.3723, 0.0000],
[ 5.3723, 0.0000]])
特征向量:
tensor([[-0.8246, -0.4159],
[ 0.5658, -0.9094]])
Other Extensions