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)