jax.lax.cumprod# jax.lax.cumprod(operand, axis=0, reverse=False)[源代码][源代码]# 计算沿 轴 的累积乘积。 参数: operand (Array) axis (int) reverse (bool) 返回类型: Array