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.

rint(x, /, out=None, *, where=True, casting=’same_kind’, order=’K’, dtype=None, subok=True[, signature, extobj])

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

See also

ceil(), floor(), trunc()

Examples

>>> a = np.array([-1.7, -1.5, -0.2, 0.2, 1.5, 1.7, 2.0])
>>> np.rint(a)
array([-2., -2., -0.,  0.,  2.,  2.,  2.])