code: def get_axes(layout, axis_names):
      axes = []
      for axis_name in axis_names:
          axis_idx = layout.find(axis_name)
          assert axis_idx >= 0
          axes.append(axis_idx)
      return axes
  
imports: from nvidia.dali.pipeline import Pipeline
  import nvidia.dali.ops
  import numpy
  from functools import partial
  from test_utils import compare_pipelines
  from test_utils import RandomDataIterator
label_desc: Use image node_id 1 for calling a global function with example usage: get_axes(layout, axis_names) and returns: axes