code: def count_nonzero(
      a: ArrayLike, axis: AxisLike = None, *, keepdims: KeepDims = False
  ):
      return a.count_nonzero(axis)
  
imports: from __future__ import annotations
  import functools
  from typing import Optional
  import torch
  from .None import _dtypes_impl, _util
  from ._normalizations import ArrayLike, AxisLike, DTypeLike, KeepDims, NotImplementedType, OutArray
label_desc: Use image node_id 3 for calling a global function with example usage: count_nonzero(a, axis) and returns: a