jax.numpy.spacing

jax.numpy.spacing(x, /)

傳回 x 和下一個相鄰數字之間的間隔。

numpy.spacing() 的 JAX 實作。

參數:

x (ArrayLike) – 實值陣列。整數或布林型別將會轉換為浮點數。

傳回:

與 x 形狀相同的陣列，包含 x 的每個條目及其最接近的相鄰值之間的間隔。

傳回型別:

陣列

參見

• jax.numpy.nextafter()：尋找下一個可表示的值。

範例

>>> x = jnp.array([0.0, 0.25, 0.5, 0.75, 1.0], dtype='float32')
>>> jnp.spacing(x)
Array([1.4012985e-45, 2.9802322e-08, 5.9604645e-08, 5.9604645e-08,
      1.1920929e-07], dtype=float32)

對於 x = 1，間隔等於 jax.numpy.finfo 給出的 eps 值

>>> x = jnp.float32(1)
>>> jnp.spacing(x) == jnp.finfo(x.dtype).eps
Array(True, dtype=bool)