code: def _canonicalize_axis(axis, rank):
      return _canonicalize_axes([axis], rank)[0]
  
imports: import funcsigs
  import numpy
  import tensorflow.compat.v2
  from trax.tf_numpy.numpy_impl import arrays
  from trax.tf_numpy.numpy_impl import dtypes
label_desc: Use image node_id 1 for calling a global function with example usage: _canonicalize_axis(axis, rank) and returns: _canonicalize_axes