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)