src/transformers/models/realm/retrieval_realm.py/ScaNNSearcher/search_batched
class ScaNNSearcher:    def search_batched(self, question_projection):
        retrieved_block_ids, _ = self.searcher.search_batched(question_projection.detach().cpu())
        return retrieved_block_ids.astype("int64")