broadcast¶
dragon.vm.torch.distributed.
broadcast
(
tensor,
src=0,
group=None
)[source]¶Broadcast the tensor from source node in a group.
- Parameters:
- tensor (dragon.vm.torch.Tensor) – The tensor to be sent.
- src (int) – The rank of the source node.
- group (ProcessGroup, optional) – The group for communication.
- Returns:
dragon.vm.torch.Tensor – The output tensor.