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)