jax.Array.cumsum

abstract Array.cumsum(axis=None, dtype=None, out=None)

返回数组的累积和。

请参考 jax.numpy.cumsum() 获取完整的文档。

参数：

• self (Array)

• axis (reductions.Axis)

• dtype (DTypeLike | None)

• out (None)

返回类型：

Array