jax.numpy.signbit

jax.numpy.signbit(x, /)

返回数组元素的符号位。

JAX 对 numpy.signbit 的实现。

参数:

x (ArrayLike) – 输入数组。不支持复数值。

返回:

一个布尔数组，形状与 x 相同，当 x 的符号为负时包含 True，否则包含 False。

返回类型:

Array

另请参阅

• jax.numpy.sign(): 返回数组元素的数学符号，即 -1、0 或 +1。

示例

布尔值上的 signbit() 始终为 False

>>> x = jnp.array([True, False])
>>> jnp.signbit(x)
Array([False, False], dtype=bool)

整数值上的 signbit() 等价于 x < 0

>>> x = jnp.array([-2, -1, 0, 1, 2])
>>> jnp.signbit(x)
Array([ True,  True, False, False, False], dtype=bool)

浮点值上的 signbit() 返回浮点表示中的实际符号位的数值，包括带符号的零

>>> x = jnp.array([-1.5, -0.0, 0.0, 1.5])
>>> jnp.signbit(x)
Array([ True, True, False, False], dtype=bool)

这也返回特殊值（如带符号的 NaN 和带符号的无穷大）的符号位

>>> x = jnp.array([jnp.nan, -jnp.nan, jnp.inf, -jnp.inf])
>>> jnp.signbit(x)
Array([False,  True, False,  True], dtype=bool)