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

torch.distributed.broadcast

torch.distributed.broadcast(tensor, src, group=None, async_op=False)

paddle.distributed.broadcast

paddle.distributed.broadcast(tensor, src, group=0)

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

参数映射

PyTorch PaddlePaddle 备注
tensor tensor 如果当前进程编号是源,那么这个 Tensor 变量将被发送给其他进程,否则这个 Tensor 将接收源发送过来的数据。
src src 发送源的进程编号。
group group 工作的进程组编号。
async_op - 是否异步操作,Paddle 无此参数,暂无转写方式。