megengine.functional.distributed.remote_recv

remote_recv(src_rank, device=None, inp=None)[源代码]

从远端进程接收一个张量。

参数
  • src_rank (int) – 源进程的序号。

  • device (Optional[str]) – 被接收的张量将要放置的设备。

  • inp – 虚拟输入以确定接收张量类型

返回类型

Tensor