mlx.core.all

目錄

mlx.core.all#

all(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) array#

在指定軸上進行 and 歸約。

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

  • axis (int or list(int), optional) -- 可選的歸約軸或多個軸。若未指定,預設對整個陣列歸約。

  • keepdims (bool, optional) -- 將被歸約的軸保留為單一維度,預設為 False

回傳:

對應軸已歸約的輸出陣列。

回傳型別:

array