jax.numpy.bitwise_count

jax.numpy.bitwise_count(x, /)

计算 x 的每个元素的绝对值的二进制表示中 1 的位数。

JAX 实现 numpy.bitwise_count。

参数：

x (ArrayLike) – 输入数组，仅接受整数子类型

返回：

一个类似数组的对象，包含 x 中每个元素的绝对值的二进制 1 位计数，与 x 形状相同，数据类型为 uint8。

返回类型：

数组

示例

>>> x1 = jnp.array([64, 32, 31, 20])
>>> # 64 = 0b1000000, 32 = 0b100000, 31 = 0b11111, 20 = 0b10100
>>> jnp.bitwise_count(x1)
Array([1, 1, 5, 2], dtype=uint8)

>>> x2 = jnp.array([-16, -7, 7])
>>> # |-16| = 0b10000, |-7| = 0b111, 7 = 0b111
>>> jnp.bitwise_count(x2)
Array([1, 3, 3], dtype=uint8)

>>> x3 = jnp.array([[2, -7],[-9, 7]])
>>> # 2 = 0b10, |-7| = 0b111, |-9| = 0b1001, 7 = 0b111
>>> jnp.bitwise_count(x3)
Array([[1, 3],
       [2, 3]], dtype=uint8)