jax.numpy.signbit

jax.numpy.signbit(x, /)

返回数组元素的符号位。

numpy.signbit 的 JAX 实现。

参数:

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

返回值:

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

返回类型:

数组

另请参阅

• 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)