jax.lax.round#

jax.lax.round(x, rounding_method=RoundingMethod.AWAY_FROM_ZERO)[source]#

Elementwise round.

Rounds values to the nearest integer.

Parameters
Return type

Array

Returns

An array containing the elementwise rounding of x.