jax.numpy.linalg.pinv

目录

jax.numpy.linalg.pinv#

jax.numpy.linalg.pinv(a, rtol=None, hermitian=False, *, rcond=Deprecated)[源代码][源代码]#

计算矩阵的 (Moore-Penrose) 伪逆。

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

参数:

  • a (ArrayLike) – 形状为 (..., M, N) 的数组，包含要伪逆的矩阵。

  • rtol (ArrayLike | None) – float 或形状为 a.shape[:-2] 的类似数组。指定小奇异值的截止值。形状为 (...,)。小奇异值的截止值；小于 rtol * 最大奇异值 的奇异值被视为零。默认值根据 dtype 的浮点精度确定。

  • hermitian (bool) – 如果为真，则假设输入是厄米特矩阵，并使用更高效的算法（默认：False）

  • rcond (ArrayLike | DeprecatedArg | None) – rtol 参数的已弃用别名。如果使用，将导致 DeprecationWarning。

返回:

形状为 (..., N, M) 的数组，包含 a 的伪逆。

返回类型:

Array

参见

备注

jax.numpy.linalg.prng() 与 numpy.linalg.prng() 在 rcond 的默认值上有所不同：在 NumPy 中，默认值是 1e-15。在 JAX 中，默认值是 10. * max(num_rows, num_cols) * jnp.finfo(dtype).eps。

示例

>>> a = jnp.array([[1, 2],
...                [3, 4],
...                [5, 6]])
>>> a_pinv = jnp.linalg.pinv(a)
>>> a_pinv  
Array([[-1.333332  , -0.33333257,  0.6666657 ],
       [ 1.0833322 ,  0.33333272, -0.41666582]], dtype=float32)

伪逆作为乘法逆运算，只要输出不是秩亏的：

>>> jnp.allclose(a_pinv @ a, jnp.eye(2), atol=1E-4)
Array(True, dtype=bool)
目录