code: def _remove_axis(shape, axis):
      return shape[:axis] + shape[axis + 1 :]
  
imports: import operator
  import ivy
  from ivy.func_wrapper import with_unsupported_dtypes, with_supported_dtypes
  from ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back, handle_jax_dtype
label_desc: Use image node_id 2 for calling a global function with example usage: _remove_axis(shape, axis) and returns: unknown