o ְiS@s*ddlmZdddZeddddZdS) )_coreNcCs t||dS)aCounts the number of non-zero values in the array. .. note:: :func:`numpy.count_nonzero` returns `int` value when `axis=None`, but :func:`cupy.count_nonzero` returns zero-dimensional array to reduce CPU-GPU synchronization. Args: a (cupy.ndarray): The array for which to count non-zeros. axis (int or tuple, optional): Axis or tuple of axes along which to count non-zeros. Default is None, meaning that non-zeros will be counted along a flattened version of ``a`` Returns: cupy.ndarray of int: Number of non-zero values in the array along a given axis. Otherwise, the total number of non-zero values in the array is returned. )axis)_count_nonzero)arrG/home/ubuntu/.local/lib/python3.10/site-packages/cupy/_sorting/count.py count_nonzeros rcupy_count_nonzero)z?->lzB->lzh->lzH->lzi->lzI->lzl->lzL->lzq->lzQ->lze->lzf->lzd->lzF->lzD->l)zin0 != type_in0_raw(0)za + bzout0 = aN)N)cupyrrcreate_reduction_funcrrrrrs