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 or tuple, optional) – 沿其计数非零值的轴或轴的元组。默认为 None，表示将在 a 的展平版本上计数非零值。

返回:

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

返回类型:

cupy.ndarray of int