jax.lax.round

jax.lax.round(x, rounding_method=RoundingMethod.AWAY_FROM_ZERO)

元素级舍入。

将值舍入到最接近的整数。

参数:

• x (ArrayLike) – 要舍入的数组或标量值。

• rounding_method (RoundingMethod) – 当对中间值（例如，0.5）进行舍入时使用的方法。有关可能的值，请参阅 jax.lax.RoundingMethod。

返回:

一个包含 x 逐元素舍入结果的数组。

返回类型:

数组