jax.numpy.floor

jax.numpy.floor(x, /)

向下舍入输入到最接近的整数。

JAX 实现 numpy.floor。

参数：

x (ArrayLike) – 输入数组或标量。不能具有复数数据类型。

返回值：

一个与 x 形状和数据类型相同的数组，包含舍入到小于或等于自身值的最近整数的值。

返回类型：

数组

参见

• jax.numpy.fix()：将输入舍入到最接近零的整数。

• jax.numpy.trunc()：将输入舍入到最接近零的整数。

• jax.numpy.ceil()：向上舍入输入到最接近的整数。

示例

>>> key = jax.random.key(42)
>>> x = jax.random.uniform(key, (3, 3), minval=-5, maxval=5)
>>> with jnp.printoptions(precision=2, suppress=True):
...     print(x)
[[ 1.44 -1.77 -3.07]
 [ 3.86  2.25 -3.08]
 [-1.55 -2.48  1.32]]
>>> jnp.floor(x)
Array([[ 1., -2., -4.],
       [ 3.,  2., -4.],
       [-2., -3.,  1.]], dtype=float32)