deepfold.distributed.model_parallel.gather

deepfold.distributed.model_parallel.gather(tensor: Tensor, dim: int, bwd: str = 'split') → Tensor