mlx.core.distributed.all_min

目錄

mlx.core.distributed.all_min#

all_min(x: array, *, group: Group | None = None, stream: None | Stream | Device = None) array#

All reduce min.

Find the minimum of the x arrays from all processes in the group.

參數:
  • x (array) -- 輸入陣列。

  • group (Group) -- The group of processes that will participate in the reduction. 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 minimum of all x arrays.

回傳值型別:

array