`i ddlZddlmZejddddejddddd Zdd Zejd d ddejd d ddd ZddZdS)N)_corezraw T a, raw int32 a_sizez uint8 packedzfor (int j = 0; j < 8; ++j) { int k = i * 8 + j; int bit = k < a_size && a[k] != 0; packed |= bit << (7 - j); }cupy_packbits_bigzfor (int j = 0; j < 8; ++j) { int k = i * 8 + j; int bit = k < a_size && a[k] != 0; packed |= bit << j; }cupy_packbits_littlebiglittlerc@|jjdvrtd|td|dvrt d|}|jdzdz}tj|ftj }t|||j|S) aPacks the elements of a binary-valued array into bits in a uint8 array. This function currently does not support ``axis`` option. Args: a (cupy.ndarray): Input array. axis (int, optional): Not supported yet. bitorder (str, optional): bit order to use when packing the array, allowed values are `'little'` and `'big'`. Defaults to `'big'`. Returns: cupy.ndarray: The packed array. .. note:: When the input array is empty, this function returns a copy of it, i.e., the type of the output array is not necessarily always uint8. This exactly follows the NumPy's behaviour (as of version 1.11), alghough this is inconsistent to the documentation. .. seealso:: :func:`numpy.packbits` biuz7Expected an input array of integer or boolean data typeN axis option is not supported yetr)bitorder must be either 'big' or 'little'dtype) rkind TypeErrorNotImplementedError ValueErrorravelsizecupyzerosuint8_packbits_kernel)aaxisbitorder packed_sizepackeds h/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_binary/packing.pypackbitsr!s, w|5   EGG G !"DEEE(((DEEE  A6A:!#K Zdj 9 9 9F H %a 8 88z raw uint8 az T unpackedz)unpacked = (a[i / 8] >> (7 - i % 8)) & 1;cupy_unpackbits_bigz%unpacked = (a[i / 8] >> (i % 8)) & 1;cupy_unpackbits_littlec|jtjkrtd|t d|dvrt dtj|jdztj}t|||S)aUnpacks elements of a uint8 array into a binary-valued output array. This function currently does not support ``axis`` option. Args: a (cupy.ndarray): Input array. bitorder (str, optional): bit order to use when unpacking the array, allowed values are `'little'` and `'big'`. Defaults to `'big'`. Returns: cupy.ndarray: The unpacked array. .. seealso:: :func:`numpy.unpackbits` z2Expected an input array of unsigned byte data typeNr rr rr) rrrrrrndarrayr_unpackbits_kernel)rrrunpackeds r unpackbitsr)Os w$*LMMM !"DEEE(((DEEE|QVaZ ;;;H h '8 4 44r")Nr)rrElementwiseKernelrr!r'r)r"r r,s  #5 "#^    &e%#^  ,#9#9#9#9N #5 "|3   &e%|/   555555r"