cupy.linalg.eigh

cupy.linalg.eigh(a, UPLO='L')[source]

Eigenvalues and eigenvectors of a symmetric matrix.

This method calculates eigenvalues and eigenvectors of a given symmetric matrix.

Note

Currenlty only 2-D matrix is supported.

Note

CUDA >=8.0 is required.

Parameters:
  • a (cupy.ndarray) – A symmetric 2-D square matrix.
  • UPLO (str) – Select from 'L' or 'U'. It specifies which part of a is used. 'L' uses the lower triangular part of a, and 'U' uses the upper triangular part of a.
Returns:

Returns a tuple (w, v). w contains eigenvalues and v contains eigenvectors. v[:, i] is an eigenvector corresponding to an eigenvalue w[i].

Return type:

tuple of ndarray