code: def _count(a, axis=None):
      if axis is None:
          return a.size
      else:
          return a.shape[axis]
  
file_path: mars/mars/tensor/stats/power_divergence.py
imports: from collections import namedtuple
  import numpy
  from ...core import ExecutableTuple
  from ...utils import require_not_none
  from ..None import special
  from ..datasource import asarray
label_desc: Use image node_id 1 for calling a global function with example usage: _count(a, axis) and returns: a, a