jax.lax.linalg.svd

jax.lax.linalg.svd(x, *, full_matrices=True, compute_uv=True)

Singular value decomposition.

Returns the singular values if compute_uv is False, otherwise returns a triple containing the left singular vectors, the singular values and the adjoint of the right singular vectors.

Parameters:

• x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

• full_matrices (bool) –

• compute_uv (bool) –

Return type:

Union[Array, tuple[Array, Array, Array]]