jax.Array.cumsum#

抽象 Array.cumsum(axis=None, dtype=None, out=None)[源代码]#

返回数组的累积和。

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

参数:

  • self (Array)

  • axis (reductions.Axis)

  • dtype (DTypeLike | None)

  • out (None)

返回类型:

Array