jax.numpy.fmod

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

計算元素級浮點數模數運算。

numpy.fmod 的 JAX 實作。

參數:

• x1 (ArrayLike) – 純量或陣列。指定被除數。

• x2 (ArrayLike) – 純量或陣列。指定除數。 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)