examples/research_projects/rag/distributed_pytorch_retriever.py/RagPyTorchDistributedRetriever/_scattered
class RagPyTorchDistributedRetriever:    def _scattered(self, scatter_list, target_shape, target_type=torch.float32):
        target_tensor = torch.empty(target_shape, dtype=target_type)
        dist.scatter(target_tensor, src=0, scatter_list=scatter_list, group=self.process_group)
        return target_tensor