jax.numpy.packbits

jax.numpy.packbits(a, axis=None, bitorder='big')

将二值数组的元素打包成 uint8 数组中的位。

LAX-backend 实现的 numpy.packbits()。

原始文档字符串如下。

结果通过在末尾插入零位来填充到完整的字节。

参数:

• a (array_like) – 一个由整数或布尔值组成的数组，其元素应按位打包。

• axis (int, optional) – 进行位打包的维度。None 表示对展平后的数组进行打包。

• bitorder ({'big', 'little'}, optional) – 输入位的顺序。’big’ 将模仿 bin(val)，[0, 0, 0, 0, 0, 0, 1, 1] => 3 = 0b00000011，’little’ 将反转顺序，所以 [1, 1, 0, 0, 0, 0, 0, 0] => 3。默认为 ‘big’。

返回:

packed – 类型为 uint8 的数组，其元素表示与输入元素的逻辑（0 或非零）值相对应的位。packed 的形状与输入具有相同的维度数量（除非 axis 为 None，在这种情况下输出为一维）。

返回类型:

ndarray