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][源代码][源代码]#
- 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，则返回奇异值，否则返回包含左奇异向量、奇异值和右奇异向量共轭转置的三元组。