jax.Array.cumprod

abstract Array.cumprod(axis=None, dtype=None, out=None)

Return the cumulative product of the array.

Refer to jax.numpy.cumprod() for the full documentation.

Parameters:

• self (Array)

• axis (int | None)

• dtype (DTypeLike | None)

• out (None)

Return type:

Array