jax.numpy.round

目录

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 相同。

返回类型:

Array

备注

jnp.round 对于恰好介于四舍五入小数之间的值，会四舍五入到最接近的偶数整数。

参见

示例

>>> 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)
目录