jax.numpy.floor_divide

jax.numpy.floor_divide#

jax.numpy.floor_divide(x1, x2, /)[source]#

逐元素计算 x1 除以 x2 的向下取整除法

JAX 对 numpy.floor_divide 的实现。

参数：:

  • **x1** ( *ArrayLike*) – 输入数组，被除数

  • **x2** ( *ArrayLike*) – 输入数组，除数

返回：:

一个数组状对象，包含每个商向下取整到最接近的整数，朝向负无穷大。这等效于 Python 中的 x1 // x2。

返回类型：:

数组

注意

对于数组 x1 和 x2，x1 // x2 等效于 jnp.floor_divide(x1, x2)。

另请参阅

jax.numpy.divide() 和 jax.numpy.true_divide() 用于浮点除法。

示例

>>> x1 = jnp.array([10, 20, 30])
>>> x2 = jnp.array([3, 4, 7])
>>> jnp.floor_divide(x1, x2)
Array([3, 5, 4], dtype=int32)

>>> x1 = jnp.array([-5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5])
>>> x2 = 3
>>> jnp.floor_divide(x1, x2)
Array([-2, -2, -1, -1, -1,  0,  0,  0,  1,  1,  1], dtype=int32)

>>> x1 = jnp.array([6, 6, 6], dtype=jnp.int32)
>>> x2 = jnp.array([2.0, 2.5, 3.0], dtype=jnp.float32)
>>> jnp.floor_divide(x1, x2)
Array([3., 2., 2.], dtype=float32)
目录