jax.Array.cumprod

Contents

jax.Array.cumprod#

abstract Array.cumprod(axis=None, dtype=None, out=None)[source]#

Return the cumulative product of the array.

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

Parameters:
  • self (Array)

  • axis (reductions.Axis)

  • dtype (DTypeLike | None)

  • out (None)

Return type:

Array