jax.numpy.signbit

jax.numpy.signbit(x, /)

返回数组元素的符号位。

JAX实现的 numpy.signbit。

参数:

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)