YouTip LogoYouTip

Pytorch Torch Linalg Eigh

# PyTorch torch.linalg.eigh Function * * * [![Image 3: Pytorch torch Reference Manual]( Pytorch torch Reference Manual]( `torch.linalg.eigh` is a function in PyTorch's linear algebra module used for computing the eigendecomposition of Hermitian matrices (symmetric or complex conjugate matrices). Compared to general eigendecomposition, it is more efficient and numerically more stable. ### Function Definition torch.linalg.eigh(A, UPLO='L', out=None) **Parameters**: * `A` (Tensor): Input Hermitian matrix. * `UPLO` (str, optional): 'L' for lower triangular, 'U' for upper triangular. Default is 'L'. * `out` (tuple, optional): Output tuple. **Returns**: * `tuple`: Returns a tuple of (eigenvalues, eigenvectors). * * * ## Usage Examples ## Example import torch # Create symmetric matrix A = torch.tensor([[2.0,1.0], [1.0,2.0]]) # Hermitian eigendecomposition eigenvalues, eigenvectors = torch.linalg.eigh(A) print("Matrix A:") print(A) print("nEigenvalues:") print(eigenvalues) print("nEigenvectors:") print(eigenvectors) Output: Matrix A: tensor([[2., 1.], [1., 2.]])Eigenvalues: tensor([1., 3.])Eigenvectors: tensor([[-0.7071, 0.7071], [ 0.7071, 0.7071]]) * * * [![Image 4: Pytorch torch Reference Manual]( Pytorch torch Reference Manual](
← Pytorch Torch Linalg InvPytorch Torch Linalg Det β†’