[torch 参数更多]torch.distributed.reduce

torch.distributed.reduce

torch.distributed.reduce(tensor, dst, op=<torch.distributed.distributed_c10d.ReduceOp object>, group=None, async_op=False)

paddle.distributed.reduce

paddle.distributed.reduce(tensor, dst, op=ReduceOp.SUM, group=0)

其中 PyTorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
tensor tensor 操作的输入 Tensor。
dst dst 返回操作结果的目标进程编号。
op op 归约的具体操作。
group group 工作的进程组编号。
async_op - 是否异步操作,Paddle 无此参数,暂无转写方式。