jax.numpy.rint

jax.numpy.rint(x)[source]

Round elements of the array to the nearest integer.

LAX-backend implementation of 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