cupy.count_nonzero#

cupy.count_nonzero(a, axis=None)[source]#

Counts the number of non-zero values in the array.

Note

numpy.count_nonzero() returns int value when axis=None, but cupy.count_nonzero() returns zero-dimensional array to reduce CPU-GPU synchronization.

Parameters
  • 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

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