jax.numpy.linalg.svdvals

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

計算矩陣的奇異值。

JAX 實作的 numpy.linalg.svdvals()。

參數:

x (ArrayLike) – 形狀為 (..., M, N) 的陣列，將為其計算奇異值。

傳回:

形狀為 (..., K) 的奇異值陣列，其中 K = min(M, N)。

傳回類型:

陣列

另請參閱

jax.numpy.linalg.svd()：計算奇異值和奇異向量

範例

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