jax.lax.round

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

逐元素取整。

将数值四舍五入到最近的整数。

参数:

• x (ArrayLike) – 要取整的数组或标量值。

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

返回:

一个包含 x 逐元素取整结果的数组。

返回类型:

Array