countNonZero function

VARP countNonZero(
  1. VARP x, {
  2. List<int> axis = const [],
  3. bool keepDims = false,
})

Implementation

VARP countNonZero(VARP x, {List<int> axis = const [], bool keepDims = false}) {
  final mask = F.notEqual(x, F.scalar(0, dtype: x.dtype));
  return F.reduceSum(mask, axis: axis, keepDims: keepDims);
}