jax.numpy.remainder

jax.numpy.remainder(x1, x2, /)

返回元素按位除法的余数。

JAX 对 numpy.remainder 的实现。

参数:

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

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

返回:

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

返回类型:

Array

注意

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

另请参阅

• jax.numpy.mod(): 返回元素按位除法的余数。

• jax.numpy.fmod(): 计算元素按位的浮点数模运算。

• jax.numpy.divmod(): 计算 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)