jax.numpy.round

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

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

numpy.round() 的 JAX 实现。

参数:

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

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

• out (None) – JAX 未使用。

返回:

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