jax.lax.round#
- jax.lax.round(x, rounding_method=RoundingMethod.AWAY_FROM_ZERO)[源代码][源代码]#
逐元素四舍五入。
将值四舍五入到最近的整数。
- 参数:
x (ArrayLike) – 要四舍五入的数组或标量值。
rounding_method (RoundingMethod) – 用于四舍五入中间值（例如 0.5）的方法。有关可能的值列表，请参见
lax.RoundingMethod。
- 返回:
包含 x 元素逐个取整的数组。
- 返回类型: