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