Warning

This page was created from a pull request.

jax.lax.round

jax.lax.round(x)

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