jax.lax.round

jax.lax.round(x)[source]

Elementwise round.

Rounds values to the nearest integer. Halfway values (e.g., 0.5) are rounded away from zero.

Parameters

x (Any) –

Return type

Any