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