jax.numpy.nextafter

内容

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)
内容