idxs = idxs[None, :] if np.ndim(idxs) == 1 else idxs