jax.numpy.round

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

将输入值舍入到指定的小数位数。

JAX 对 numpy.round() 的实现。

参数:

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

• decimals (int) – 整数，默认为 0。输入需要舍入到的小数位数。必须静态指定。不支持 decimals < 0。

• out (None) – JAX 未使用。

返回:

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

返回类型:

Array

注意

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)