jax.lax.linalg.svd

Contents

jax.lax.linalg.svd#

jax.lax.linalg.svd(x, *, full_matrices=True, compute_uv=True, subset_by_index=None)[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.

Parameters:
  • x (ArrayLike) –

  • full_matrices (bool) –

  • compute_uv (bool) –

  • subset_by_index (tuple[int, int] | None) –

Return type:

Array | tuple[Array, Array, Array]