jax.lax.nextafter

内容

jax.lax.nextafter#

jax.lax.nextafter(x1, x2)[source]#

返回 x1 方向上朝 x2 方向的下一个可表示的值。

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

>>> jnp.nextafter(0, 1)  # denormal numbers are representable
Array(1.e-45, dtype=float32, weak_type=True)
>>> jnp.nextafter(0, 1) * 1  # but are flushed to zero
Array(0., dtype=float32, weak_type=True)

对于最小的可用(即正常)浮点数,请使用 tiny of jnp.finfo

参数:
  • x1 (ArrayLike)

  • x2 (ArrayLike)

返回类型:

数组