cupy.count_nonzero

cupy.count_nonzero(a, axis=None)

计算数组中非零元素的数量。

注意

numpy.count_nonzero() 在 axis=None 时返回 int 值，但 cupy.count_nonzero() 返回零维数组以减少 CPU-GPU 同步。

参数:

• a (cupy.ndarray) – 要计算非零元素的数组。

• axis (int 或 tuple, 可选) – 计算非零元素的轴或轴的元组。默认为 None，表示将沿 a 的展平版本计算非零元素。

返回值:

数组沿给定轴的非零元素数量。否则，返回数组中非零元素的总数。

返回类型:

cupy.ndarray of int