mlx.core.cumprod

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

Return the cumulative product of the elements along the given axis.

Parameters:

• a (array) – Input array

• axis (int, optional) – Optional axis to compute the cumulative product over. If unspecified the cumulative product of the flattened array is returned.

• reverse (bool) – Perform the cumulative product in reverse.

• inclusive (bool) – The i-th element of the output includes the i-th element of the input.