megengine.functional.distributed.reduce_sum

reduce_sum(inp, group=WORLD, device=None)[源代码]

创建用于聚合通信的 reduce_sum 算子

参数
返回类型

Tensor