- cupy.count_nonzero(a, axis=None)[source]#
Counts the number of non-zero values in the array.
numpy.count_nonzero()returns int value when axis=None, but
cupy.count_nonzero()returns zero-dimensional array to reduce CPU-GPU synchronization.
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
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.
- Return type
cupy.ndarray of int