code: def fix_avg_mean(avg_mean, groups):
      if avg_mean.ndim == 2:  # OK
          return avg_mean
      elif avg_mean.ndim == 1:  # Issue #7706
          if groups != 1:
              _warn_old_model()
          return _broadcast_to(avg_mean, (groups,) + avg_mean.shape)
      raise ValueError("unexpected shape of avg_mean")
  
imports: import functools
  import warnings
  import numpy
  import chainer
  from chainer import configuration
  from chainer import functions
  from chainer import link
  import chainer.serializer
  from chainer.utils import argument
label_desc: Use image node_id 2 for calling a global function with example usage: fix_avg_mean(avg_mean, groups) and returns: avg_mean, _broadcast_to