megengine.functional.distributed.all_reduce_max

all_reduce_max(inp, group=WORLD, device='')[源代码]

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

参数
  • inp – 输入张量。

  • group – 通信组。

  • device – 执行设备。