jax.numpy.remainder#

jax.numpy.remainder(x1, x2, /)[源代码]#

返回除法的按元素余数。

numpy.remainder 的 JAX 实现。

参数:
  • x1 (ArrayLike) – 标量或数组。指定被除数。

  • x2 (ArrayLike) – 标量或数组。指定除数。 x1x2 应具有相同的形状或广播兼容。

返回:

一个数组,其中包含 x1 按元素除以 x2 的余数,其符号与 x2 的元素相同。

返回类型:

Array

注意

jnp.remainder 的结果等价于 x1 - x2 * jnp.floor(x1 / x2)

另请参阅

示例

>>> x1 = jnp.array([[3, -1, 4],
...                 [8, 5, -2]])
>>> x2 = jnp.array([2, 3, -5])
>>> jnp.remainder(x1, x2)
Array([[ 1,  2, -1],
       [ 0,  2, -2]], dtype=int32)
>>> x1 - x2 * jnp.floor(x1 / x2)
Array([[ 1.,  2., -1.],
       [ 0.,  2., -2.]], dtype=float32)