jax.numpy.unpackbits

内容

jax.numpy.unpackbits#

jax.numpy.unpackbits(a, axis=None, count=None, bitorder='big')[source]#

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

LAX 后端实现 numpy.unpackbits().

原始文档字符串如下。

每个 `a` 元素代表一个位域，应解包到二进制值输出数组中。输出数组的形状要么是 1 维（如果 `axis` 为 `None`），要么与输入数组的形状相同，解包沿指定的轴进行。

参数：:

  • **a** (ndarray, uint8 类型) – 输入数组。

  • axis (int, 可选) – 位解包操作的维度。 None 表示解包扁平化后的数组。

  • count (int 或 None, 可选) – 沿 axis 解包的元素数量，作为撤消对非八位数倍数大小进行打包的效果的一种方式。非负数表示仅解包 count 位。负数表示从末尾截断这么多位。 None 表示解包整个数组（默认值）。大于可用位数的计数将对输出添加零填充。负计数不能超过可用位数。

  • bitorder ({'big', 'little'}, 可选) – 返回位的顺序。'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 类型

内容