deepfold.distributed.model_parallel.transpose_on_model_parallel_region

deepfold.distributed.model_parallel.transpose_on_model_parallel_region(tensor: Tensor, dim0: int, dim1: int) Tensor[source]