jax.lax.nextafter#

jax.lax.nextafter(x1, x2)[源代码]#

返回 x1 之后，在 x2 方向上的下一个可表示值。

此函数直接降到 chlo.next_after 操作。

参数:

  • x1 (ArrayLike) – 输入数组。必须具有匹配的浮点 dtype。如果两者都不是标量，则必须具有相同的维度数并且是广播兼容的。

  • x2 (ArrayLike) – 输入数组。必须具有匹配的浮点 dtype。如果两者都不是标量，则必须具有相同的维度数并且是广播兼容的。

返回:

包含 x1 之后，在 x2 方向上的下一个可表示浮点值的，与输入具有相同 dtype 和广播形状的数组。

返回类型:

数组

注意事项

在某些环境中，使用 flush-denormal-to-zero 语义。这意味着，在零附近，此函数返回严格非零值，这些值在任何操作中都显示为零。考虑这个例子

>>> from jax import lax
>>> lax.nextafter(0.0, 1.0)  # denormal numbers are representable
Array(1.e-45, dtype=float32, weak_type=True)
>>> lax.nextafter(0.0, 1.0) * 1  # but are flushed to zero
Array(0., dtype=float32, weak_type=True)

对于最小可用（即正常）的浮点数，请使用 jnp.finfo 的 tiny。