megengine.functional.distributed.remote_recv

remote_recv(src_rank, shape, dtype, device=None, inp=None)

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

参数

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

• shape (Tuple[int]) – 被接收的张量的形状。

• dtype (type) – 被接收的张量的数据类型。

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

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

返回类型

Tensor

megengine.functional.distributed.remote_send megengine.functional.distributed.all_gather