def _gather_nd(params, indices, batch_dims=0, name=None):
    'gather_nd.'
    indices = ops.convert_to_tensor(indices, dtype_hint=np.int32)
    if (batch_dims < 0):
        raise NotImplementedError('Negative `batch_dims` is currently unsupported.')
    if ((not JAX_MODE) and (batch_dims > 0)):
        raise NotImplementedError('`batch_dims > 0` currently unsupported in NumPy backend.')
    gather_nd_ = _gather_nd_single
    if JAX_MODE:
        gather_nd_ = functools.reduce((lambda g, f: f(g)), ([jax.vmap] * int(batch_dims)), gather_nd_)
    return gather_nd_(params, indices)
gather_nd.
def _gather_nd(params, indices, batch_dims=0, name=None):
    
    indices = ops.convert_to_tensor(indices, dtype_hint=np.int32)
    if (batch_dims < 0):
        raise NotImplementedError('Negative `batch_dims` is currently unsupported.')
    if ((not JAX_MODE) and (batch_dims > 0)):
        raise NotImplementedError('`batch_dims > 0` currently unsupported in NumPy backend.')
    gather_nd_ = _gather_nd_single
    if JAX_MODE:
        gather_nd_ = functools.reduce((lambda g, f: f(g)), ([jax.vmap] * int(batch_dims)), gather_nd_)
    return gather_nd_(params, indices)