megengine.functional.distributed.remote_recv

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

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

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

  • device – 被接收的张量将要放置的设备。

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