mlx.core.median

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

Compute the median(s) over the given axes.

參數:

• a (array) -- 輸入陣列。

• axis (int or list(int), optional) -- Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

• keepdims (bool, optional) -- Keep reduced axes as singleton dimensions, defaults to False.

回傳值:

中位數的輸出陣列。

回傳值型別:

array