jax.lax.round#
- jax.lax.round(x, rounding_method=RoundingMethod.AWAY_FROM_ZERO)[source]#
逐元素舍入。
将值舍入到最接近的整数。
- 参数:
x (ArrayLike) – 要舍入的数组或标量值。
rounding_method (RoundingMethod) – 在舍入半途值（例如 0.5）时使用的舍入方法。有关可能的舍入方法列表，请参见
lax.RoundingMethod。
- 返回值:
包含 x 的逐元素舍入的数组。
- 返回类型: