jax.lax.linalg.svd

jax.lax.linalg.svd(x, full_matrices=True, compute_uv=True)[source]

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.