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