jax.numpy.packbits¶

jax.numpy.packbits(a, axis=None, bitorder='big')[source]¶

Packs the elements of a binary-valued array into bits in a uint8 array.

LAX-backend implementation of packbits(). Original docstring below.

packbits(a, axis=None, bitorder=’big’)

The result is padded to full bytes by inserting zero bits at the end.
Returns
packedndarray
Array of type uint8 whose elements represent bits corresponding to the logical (0 or nonzero) value of the input elements. The shape of packed has the same number of dimensions as the input (unless axis is None, in which case the output is 1-D).

unpackbits: Unpacks elements of a uint8 array into a binary-valued output
array.
>>> a = np.array([[[1,0,1],
...                [0,1,0]],
...               [[1,1,0],
...                [0,0,1]]])
>>> b = np.packbits(a, axis=-1)
>>> b
array([[[160],
        [ 64]],
       [[192],
        [ 32]]], dtype=uint8)
Note that in binary 160 = 1010 0000, 64 = 0100 0000, 192 = 1100 0000, and 32 = 0010 0000.