jax.numpy.rint

jax.numpy.rint(x, /)

将 x 的元素四舍五入到最近的整数

JAX implementation of numpy.rint.

参数:

x (ArrayLike) – 输入数组

返回:

一个包含 x 元素四舍五入后的类数组对象。始终转换为不精确类型。

返回类型:

Array

备注

如果 x 的元素正好是半途，例如 0.5 或 1.5，rint 将四舍五入到最近的偶数整数。

示例

>>> 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)