jax.numpy.round

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

将输入均匀地舍入到给定的小数位数。

numpy.round() 的 JAX 实现。

参数:

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

• decimals (int) – 整数，默认值=0。输入需要舍入到的小数位数。它必须静态指定。未实现 decimals < 0 的情况。

• out (None) – JAX 未使用。

返回值:

一个包含舍入到指定 decimals 值的数组，其形状和 dtype 与 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)