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) tuple[Array, Array, Array][source]#
- jax.lax.linalg.svd(x: ArrayLike, *, full_matrices: bool = True, compute_uv: Literal[False], subset_by_index: tuple[int, int] | 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) Array | tuple[Array, Array, Array]
奇异值分解。
如果 compute_uv 为 False，则返回奇异值；否则返回一个三元组，包含左奇异向量、奇异值和右奇异向量的共轭转置。