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.