jax.numpy.nextafter#

jax.numpy.nextafter(x, y, /)[source]#

傳回元素級別的下一個浮點數值，從 x 朝向 y。

JAX 實現的 numpy.nextafter。

參數:

  • x (ArrayLike) – 純量或陣列。指定尋找下一個數字的起始值。

  • y (ArrayLike) – 純量或陣列。指定尋找下一個數字的方向。x 和 y 應該具有相同的形狀或可廣播相容。

傳回值:

一個陣列，包含 x 在 y 方向上的下一個可表示數字。

傳回類型:

陣列

範例

>>> jnp.nextafter(2, 1)  
Array(1.9999999, dtype=float32, weak_type=True)
>>> x = jnp.array([3, -2, 1])
>>> y = jnp.array([2, -1, 2])
>>> jnp.nextafter(x, y)  
Array([ 2.9999998, -1.9999999,  1.0000001], dtype=float32)