jax.lax.linalg.eig#
- jax.lax.linalg.eig(x, *, compute_left_eigenvectors=True, compute_right_eigenvectors=True)[source]#
Eigendecomposition of a general matrix.
Nonsymmetric eigendecomposition is at present only implemented on CPU.
- Parameters:
- Return type:
- Returns:
The eigendecomposition of
x
, which is a tuple of the form(w, vl, vr)
wherew
are the eigenvalues,vl
are the left eigenvectors, andvr
are the right eigenvectors.vl
andvr
are optional and will only be included ifcompute_left_eigenvectors
orcompute_right_eigenvectors
respectively areTrue
.If the eigendecomposition fails, then arrays full of NaNs will be returned for that batch element.