megengine.functional.distributed.remote_recv# remote_recv(src_rank, device=None, inp=None)[源代码]# 从另一个进程接收张量。 参数: src_rank (int) – 源进程的编号 device (Optional[str]) – 执行此操作的设备。默认为输入张量所在的设备。可以通过指定设备为 ”gpu0:1“ 以在不同的 cuda 流上执行此操作,其中1是 cuda 流的编号,默认 cuda 流编号为0。 inp – 确定接收张量类型的模拟输入。 返回类型: Tensor 返回: 被接收的张量。 例如: 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)