code: def _gather_nd_single(params, indices):
      idx = tuple(np.moveaxis(indices, -1, 0))
      return params[idx]
  
imports: import functools
  import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import ops
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import einsum
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import norm
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import tensordot
label_desc: Use image node_id 5 for calling a global function with example usage: _gather_nd_single(params, indices) and returns: params