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, butcupy.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