YouTip LogoYouTip

Pytorch Torch Linalg Eigvals

# PyTorch torch.linalg.eigvals Function * * Pytorch torch Reference Manual](#) `torch.linalg.eigvals` is a function in PyTorch's linear algebra module used to compute the eigenvalues of a square matrix. It returns only the eigenvalues and does not compute eigenvectors, making it more efficient than `torch.linalg.eig`. ### Function Definition torch.linalg.eigvals(A, out=None) **Parameters**: * `A` (Tensor): Input square matrix. * `out` (Tensor, optional): Output tensor. **Returns**: * `torch.Tensor`: Returns eigenvalues. * * * ## Usage Example ## Instance import torch # Create a square matrix A = torch.tensor([[1.0,2.0],[3.0,4.0]], dtype=torch.complex128) # Calculate eigenvalues eigenvalues = torch.linalg.eigvals(A) print("Matrix A:") print(A) print("nEigenvalues:") print(eigenvalues) Output: Matrix A: tensor([[1., 2.], [3., 4.]], dtype=torch.complex128)Eigenvalues: tensor([-0.3723+0.j, 5.3723+0.j], dtype=torch.complex128) * * Pytorch torch Reference Manual](#)
← Pytorch Torch Linalg LstsqPytorch Torch Linalg Eig β†’