code: def _first(arr, axis):
      """Return arr[..., 0:1, ...] where 0:1 is in the `axis` position"""
  
      return cupy.take_along_axis(
          arr, cupy.array(0, ndmin=arr.ndim), axis
      )
  
imports: import cupy
label_desc: Use image node_id 1 for calling a global function with example usage: _first(arr, axis) and returns: cupy