mlx_rs::linalg

Function eigh_device

Source
pub fn eigh_device(
    a: impl AsRef<Array>,
    uplo: Option<&str>,
    stream: impl AsRef<Stream>,
) -> Result<(Array, Array)>
Expand description

Compute the eigenvalues and eigenvectors of a complex Hermitian or real symmetric matrix.

This function supports arrays with at least 2 dimensions. When the input has more than two dimensions, the eigenvalues and eigenvectors are computed for each matrix in the last two dimensions.