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