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 (jax.typing.ArrayLike)

• decimals (int)

• out (None)

Return type:

Array