jax.lax.linalg.svd

jax.lax.linalg.svd(x: ArrayLike, *, full_matrices: bool = True, compute_uv: Literal[True], subset_by_index: tuple[int, int] | None = None, algorithm: SvdAlgorithm | None = None) → tuple[Array, Array, Array]

jax.lax.linalg.svd(x: ArrayLike, *, full_matrices: bool = True, compute_uv: Literal[False], subset_by_index: tuple[int, int] | None = None, algorithm: SvdAlgorithm | None = None) → Array

jax.lax.linalg.svd(x: ArrayLike, *, full_matrices: bool = True, compute_uv: bool = True, subset_by_index: tuple[int, int] | None = None, algorithm: SvdAlgorithm | None = None) → Array | tuple[Array, Array, Array]

奇異值分解。

如果 compute_uv 為 False，則傳回奇異值；否則傳回包含左奇異向量、奇異值和右奇異向量的共軛轉置的三元組。