megengine.functional.distributed.all_reduce_sum

all_reduce_sum(inp, group=<megengine.distributed.group.Group object>, device='')

创建用于聚合通信的 all_reduce_sum 算子。

参数

• inp (Tensor) – 输入张量。

• group (Optional[Group]) – 通信组。

• device (Optional[str]) – 执行设备。

返回类型

Tensor

megengine.functional.distributed.all_reduce_min megengine.functional.distributed.broadcast