jax.numpy.round

jax.numpy.round(a, decimals=0, out=None)

将输入均匀舍入到给定的十进制数。

JAX 实现 numpy.round()。

参数：

• a (ArrayLike) – 输入数组或标量。

• decimals (int) – int，默认为 0。输入需要舍入到的十进制位数。它必须是静态指定的。对于 decimals < 0 未实现。

• out (None) – JAX 未使用。

返回值：

一个包含舍入到指定 decimals 的值的数组，与 a 具有相同的形状和数据类型。

返回类型：

数组

注意

jnp.round 将正好位于舍入十进制值之间一半的值舍入到最接近的偶数整数。

另请参阅

• jax.numpy.floor()：将输入向下舍入到最接近的整数。

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

• jax.numpy.fix() 和 :func:numpy.trunc：向零取整到最接近的整数。

示例

>>> x = jnp.array([1.532, 3.267, 6.149])
>>> jnp.round(x)
Array([2., 3., 6.], dtype=float32)
>>> jnp.round(x, decimals=2)
Array([1.53, 3.27, 6.15], dtype=float32)

对于正好位于取整值中间的值

>>> x1 = jnp.array([10.5, 21.5, 12.5, 31.5])
>>> jnp.round(x1)
Array([10., 22., 12., 32.], dtype=float32)