jax.numpy.round

jax.numpy.round(a, decimals=0, out=None)

Round an array to the given number of decimals.

LAX-backend implementation of numpy.around().

Original docstring below.

around is an alias of ~numpy.round.

Parameters:

• a (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

• decimals (int) –

• out (None) –

Return type:

Array