jax.lax.nextafter

目录

jax.lax.nextafter#

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

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

请注意,在某些环境中使用的是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)

对于最小的可用(即正常)浮点数,使用 jnp.finfotiny

参数:
  • x1 (ArrayLike)

  • x2 (ArrayLike)

返回类型:

Array