mlx.core.cumsum

cumsum(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array

返回沿给定轴的元素的累加和。

参数：

• a (array) – 输入数组

• axis (int, optional) – 可选轴，用于计算累加和。如果未指定，则返回展平数组的累加和。

• reverse (bool) – 按反向顺序执行累加和计算。

• inclusive (bool) – 输出的第 i 个元素包含输入的第 i 个元素。

返回值：

输出数组。

返回类型：

array