mlx.core.distributed.recv

recv(shape: Sequence[int], dtype: Dtype, src: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array

從排名為 src 的程序接收形狀為 shape 且 dtype 為 dtype 的陣列。

參數:

• shape (Tuple[int]) -- The shape of the array we are receiving.

• dtype (Dtype) -- The data type of the array we are receiving.

• src (int) -- Rank of the source process in the group.

• group (Group) -- The group of processes that will participate in the recv. If set to None the global group is used. Default: None.

• stream (Stream, optional) -- Stream or device. Defaults to None in which case the default stream of the default device is used.

回傳:

The array that was received from src.

回傳型別:

array