mlx.core.sum

sum(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