jax.lax.cumsum

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

沿著 axis 計算累計總和。

參數:

• operand (Array)

• axis (int)

• reverse (bool)

回傳類型:

Array