mlx.core.distributed.send

send(x: array, dst: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array

将数组从当前进程发送到组中等级为 dst 的进程。

参数:

• x (array) – 输入数组。

• dst (int) – 组中目标进程的等级。

• group (Group) – 参与发送的进程组。如果设置为 None，则使用全局组。默认值: None。

• stream (Stream, 可选) – 流或设备。默认为 None，此时使用默认设备的默认流。

返回值:

一个与 x 相同的数组，当该数组被求值时，发送操作被执行。

返回类型:

array