megengine.functional.distributed.all_gather

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

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

参数
返回类型

Tensor