deepfold.distributed.model_parallel.scatter

deepfold.distributed.model_parallel.scatter(tensor: Tensor, dim: int) Tensor