从 rank 为 src
的进程接收一个形状和类型与 x
相似的数组。
这等价于调用 mx.distributed.recv(x.shape, x.dtype, src)
。
- 参数:
x (array) – 定义接收数组的形状和数据类型的数组。
src (int) – 组中源进程的 rank。
group (Group) – 参与接收的进程组。如果设置为 None
,则使用全局组。默认值: None
。
stream (Stream, 可选) – 流或设备。默认为 None
,在这种情况下使用默认设备的默认流。
- 返回值:
从 src
接收到的数组。
- 返回类型:
array