mlx.core.mean

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

计算给定轴上的均值。

参数:

• a (array) – 输入数组。

• axis (int _或_ list(int)_,_ _可选_) – 要执行归约的轴或多个轴。如果未指定，默认为对整个数组进行归约。

• keepdims (bool_,_ _可选_) – 保留被归约的轴作为单维，默认为 False。

返回值:

均值的输出数组。

返回类型:

array