jax.numpy.unpackbits

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

将 uint8 数组的元素解包到二进制值的输出数组中。

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

原始文档字符串如下。

a 的每个元素代表一个应解包为二值输出数组的位域。输出数组的形状为1-D（如果 axis 为 None），或者与输入数组形状相同，并在指定的轴上进行解包。

参数:

• a (ndarray, uint8 type) – 输入数组。

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

• count (int or None, optional) – 要沿 axis 解包的元素数量，提供了一种方式来撤销打包非八的倍数大小的效果。非负数表示只解包 count 位。负数表示从末尾修剪掉那么多位。None 表示解包整个数组（默认）。计数大于可用位数将在输出中添加零填充。负计数不得超过可用位数。

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

返回:

unpacked – 元素是二进制值（0 或 1）。

返回类型:

ndarray, uint8 type