send

paddle.distributed. send ( tensor, dst=0, group=None, use_calc_stream=True ) [source]

Send a tensor to the receiver.

Parameters
  • tensor (Tensor) – The Tensor to send. Its data type should be float16, float32, float64, int32 or int64.

  • dst (int) – The destination rank id.

  • group (Group, optional) – The group instance return by new_group or None for global default group. Default: None.

  • use_calc_stream (bool, optional) – Whether to use calculate stream or communication stream. Default: True.

Returns

None.

Examples

System Message: ERROR/3 (/usr/local/lib/python3.8/site-packages/paddle/distributed/collective.py:docstring of paddle.distributed.collective.send, line 17)

Error in “code-block” directive: maximum 1 argument(s) allowed, 31 supplied.

.. code-block:: python
    # required: distributed
    import paddle
    from paddle.distributed import init_parallel_env
    init_parallel_env()
    if paddle.distributed.ParallelEnv().rank == 0:
        data = paddle.to_tensor([7, 8, 9])
        paddle.distributed.send(data, dst=1)
    else:
        data = paddle.to_tensor([1,2,3])
        paddle.distributed.recv(data, src=0)
    out = data.numpy()