jax.numpy.round#

jax.numpy.round(a, decimals=0, out=None)[原始碼]#

將輸入值均勻四捨五入到給定的小數位數。

JAX 實作的 numpy.round()。

參數:

  • a (ArrayLike) – 輸入陣列或純量。

  • decimals (int) – 整數，預設值=0。輸入值需要四捨五入到的小數位數。必須靜態指定。尚未實作 decimals < 0 的情況。

  • out (None) – JAX 未使用。

傳回值:

一個陣列，包含四捨五入到指定 decimals 的值，其形狀和 dtype 與 a 相同。

傳回型別:

Array

注意

jnp.round 對於精確地在四捨五入後的小數位值之間的中途值，會四捨五入到最接近的偶數整數。

另請參閱

  • jax.numpy.floor()：將輸入值向下四捨五入到最接近的整數。

  • jax.numpy.ceil()：將輸入值向上四捨五入到最接近的整數。

  • jax.numpy.fix() 和 :func:numpy.trunc`: 將輸入值朝零的方向四捨五入到最接近的整數。

範例

>>> x = jnp.array([1.532, 3.267, 6.149])
>>> jnp.round(x)
Array([2., 3., 6.], dtype=float32)
>>> jnp.round(x, decimals=2)
Array([1.53, 3.27, 6.15], dtype=float32)

對於精確地在四捨五入值之間的中途值

>>> x1 = jnp.array([10.5, 21.5, 12.5, 31.5])
>>> jnp.round(x1)
Array([10., 22., 12., 32.], dtype=float32)