jax.lax.cumprod

jax.lax.cumprod(operand, axis=0, reverse=False)

沿着 axis 计算累积积。

参数：

• operand (Array)

• axis (int)

• reverse (bool)

返回类型：

Array