[参数不一致]torch.distributed.all_to_all

torch.distributed.all_to_all

torch.distributed.all_to_all(output_tensor_list, input_tensor_list, group=None, async_op=False)

paddle.distributed.alltoall

paddle.distributed.alltoall(in_tensor_list, out_tensor_list, group=None, sync_op=True)

其中 PyTorch 和 Paddle 功能一致,参数用法不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
output_tensor_list out_tensor_list 包含所有输出 Tensors 的一个列表,仅参数名不一致。
input_tensor_list in_tensor_list 包含所有输入 Tensors 的一个列表,仅参数名不一致。
group group new_group 返回的 Group 实例,或者设置为 None 表示默认地全局组。
async_op sync_op torch 为是否异步操作,Paddle 为是否同步操作,转写方式取反即可。