megengine.functional.distributed.all_to_all

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

创建用于聚合通信 all_to_all 算子。

参数
返回类型

Tensor