get_group

paddle.distributed. get_group ( id=0 ) [源代码]

通过通信组 id 获取通信组实例

参数

  • id (int) - 通信组 id。默认值为 0。

返回

Group 通信组实例

代码示例

>>> import paddle
>>> import paddle.distributed as dist

>>> dist.init_parallel_env()
>>> gid = paddle.distributed.new_group([2,4,6])
>>> paddle.distributed.get_group(gid.id)