jax.numpy.rint

Contents

jax.numpy.rint#

jax.numpy.rint(x, /)[source]#

Round elements of the array to the nearest integer.

LAX-backend implementation of numpy.rint().

Original docstring below.

Parameters:

x (array_like) – Input array.

Returns:

out – Output array is same shape and type as x. This is a scalar if x is a scalar.

Return type:

ndarray or scalar