jax.numpy.rint#

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

Rounds the elements of x to the nearest integer

JAX implementation ofnumpy.rint.

Parameters:

x (ArrayLike) – Input array

Returns:

An array-like object containing the rounded elements ofx. Always promotesto inexact.

Return type:

Array

Note

If an element of x is exactly half way, e.g.0.5 or1.5, rint will roundto the nearest even integer.

Examples

>>>x1=jnp.array([5,4,7])>>>jnp.rint(x1)Array([5., 4., 7.], dtype=float32)

>>>x2=jnp.array([-2.5,-1.5,-0.5,0.5,1.5,2.5,3.5,4.5])>>>jnp.rint(x2)Array([-2., -2., -0.,  0.,  2.,  2.,  4.,  4.], dtype=float32)

>>>x3=jnp.array([-2.5+3.5j,4.5-0.5j])>>>jnp.rint(x3)Array([-2.+4.j,  4.-0.j], dtype=complex64)