megengine.functional.distributed.remote_send

remote_send(inp, dest_rank)[源代码]

发送张量到另一个进程。

参数
  • inp (Tensor) – 需要发送的张量。

  • dest_rank (int) – 目标进程的编号。

返回

None.

实际案例

if rank == 0:
    data = mge.tensor(1)
    # Tensor(1)
    F.distributed.remote_send(data, 1) # return None
else:
    data = F.distributed.remote_recv(0)
    # Tensor(1)