code: def axis_indices(shape, axis):
      r = np.arange(shape[axis])
      r = np.expand_dims(
          r, list(range(0, axis)) + list(range(axis + 1, len(shape)))
      )
      rep = list(shape)
      rep[axis] = 1
      r = np.tile(r, rep)
      return r
  
imports: from nose_utils import assert_raises
  import nvidia.dali
  import nvidia.dali.fn
  import nvidia.dali.ops
  import numpy
  import scipy.ndimage
  import scipy.ndimage.measurements
  import random
  from test_utils import check_batch, np_type_to_dali
  from nose.tools import nottest
  from nose.plugins.attrib import attr
label_desc: Use image node_id 11 for calling a global function with example usage: axis_indices(shape, axis) and returns: r