code: def gather_elements(data, indices, axis=0):  # type: ignore
      data_swaped = np.swapaxes(data, 0, axis)
      index_swaped = np.swapaxes(indices, 0, axis)
      gathered = np.choose(index_swaped, data_swaped, mode="wrap")
      y = np.swapaxes(gathered, 0, axis)
      return y
  
file_path: onnx/onnx/backend/test/case/node/gatherelements.py
imports: import numpy
  import onnx
  from onnx.backend.test.case.base import Base
  from onnx.backend.test.case.node import expect
label_desc: Use image node_id 1 for calling a global function with example usage: gather_elements(data, indices, axis) and returns: y