jax.numpy.linalg.svdvals

jax.numpy.linalg.svdvals(x, /)

计算矩阵的奇异值。

numpy.linalg.svdvals() 的 JAX 实现。

参数：

**x** (ArrayLike) – 形状为 (..., M, N) 的数组，将计算其奇异值。

返回：

形状为 (..., K) 的奇异值数组，其中 K = min(M, N)。

返回类型：

Array

另请参见

jax.numpy.linalg.svd()：计算奇异值和奇异向量

示例

>>> x = jnp.array([[1, 2, 3],
...                [4, 5, 6]])
>>> jnp.linalg.svdvals(x)
Array([9.508031 , 0.7728694], dtype=float32)