jax.lax.cumprod

Contents

jax.lax.cumprod#

jax.lax.cumprod(operand, axis=0, reverse=False)[source]#

Computes a cumulative product along axis.

Parameters:
  • operand (Array) –

  • axis (int) –

  • reverse (bool) –

Return type:

Array