jax.numpy.fmod

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

计算按元素的浮点数模运算。

numpy.fmod 的 JAX 实现。

参数:

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

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

返回:

一个数组，其中包含 x1 和 x2 的按元素浮点数模运算的结果，其符号与 x1 的元素相同。

返回类型:

数组

注意

jnp.fmod 的结果等价于 x1 - x2 * jnp.fix(x1 / x2)。

另请参阅

• jax.numpy.mod() 和 jax.numpy.remainder()：返回逐元素的除法余数。

• jax.numpy.divmod()：逐元素计算 x1 除以 x2 的整数商和余数。

示例

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