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 group for communication.
Returns:

dragon.Tensor – The output tensor.