jax.lax.linalg.svd#
- jax.lax.linalg.svd(x: Array | ndarray | bool_ | number | bool | int | float | complex, *, full_matrices: bool = True, compute_uv: Literal[True], subset_by_index: tuple[int, int] | None = None) tuple[Array, Array, Array] [source]#
- jax.lax.linalg.svd(x: Array | ndarray | bool_ | number | bool | int | float | complex, *, full_matrices: bool = True, compute_uv: Literal[False], subset_by_index: tuple[int, int] | None = None) Array
- jax.lax.linalg.svd(x: Array | ndarray | bool_ | number | bool | int | float | complex, *, full_matrices: bool = True, compute_uv: bool = True, subset_by_index: tuple[int, int] | None = None) Array | tuple[Array, Array, Array]
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.