jax.numpy.packbits#
- jax.numpy.packbits(a, axis=None, bitorder='big')[source]#
将二进制值数组的元素打包到 uint8 数组中的位中。
LAX 后端实现
numpy.packbits()。原始文档字符串如下。
结果通过在末尾插入零位填充到完整的字节。
- 参数：
a (array_like) – 整数或布尔值的数组，其元素应打包到位。
axis (int, 可选) – 执行位打包的维度。
None表示打包扁平化的数组。bitorder ({'big', 'little'}, 可选) – 输入位的顺序。‘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，在这种情况下输出为 1 维）。
- 返回类型:
ndarray