jax.numpy.linalg.svdvals

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

计算矩阵的奇异值。

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

参数:

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)