mlx.core.cumprod

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