broadcast¶

dragon.distributed.broadcast(
  inputs,
  root=0,
  group=None,
  **kwargs
)[source]¶

Broadcast the input from root node in a group.

Parameters:

• inputs (dragon.Tensor) – The tensor to broadcast.
• root (int, optional, default=0) – The node index in the group.
• group (ProcessGroup, optional) – The communication group.

Returns:

dragon.Tensor – The output tensor.