def _gather_nd_single(params, indices):
    idx = tuple(np.moveaxis(indices, (- 1), 0))
    return params[idx]
def _gather_nd_single(params, indices):
    idx = tuple(np.moveaxis(indices, (- 1), 0))
    return params[idx]