jax.numpy.rint

jax.numpy.rint(x, /)

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

numpy.rint 的 JAX 实现。

参数:

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)