Are the idxs that you want to fetch contiguous ?

No. The large dataset contains pre-computed embeddings (and other labels) for all user utterances. Given any particular user utterance (or in this scenario, a particular 300k utterances), I want to be able to fetch its pre-computed embedding from the big dataset.