jax.numpy.bitwise_count

jax.numpy.bitwise_count(x, /)

計算 x 中每個元素的絕對值的二進位表示法中 1 位元的數量。

numpy.bitwise_count 的 JAX 實作。

參數:

x (ArrayLike) – 輸入陣列，僅接受整數子型別

回傳:

一個類陣列物件，包含 x 中每個元素的絕對值的二進位 1 位元計數，形狀與 x 相同，dtype 為 uint8。

回傳型別:

Array

範例

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