CumSumΒΆ

Performs cumulative sum of the input elements along the given axis. By default, it will do the sum inclusively meaning the first element is copied as is. Through an exclusive attribute, this behavior can change to exclude the first element. It can also perform summation in the opposite direction o…

Abstract Signature:

CumSum(x: Tensor, axis, exclusive: int, reverse: int)

PyTorch

API: torch.cumsum
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.cumsum
Strategy: Direct Mapping

NumPy

API: numpy.cumsum
Strategy: Direct Mapping

Keras

API: keras.ops.numpy.cumsum
Strategy: Direct Mapping

TensorFlow

API: tf.math.cumsum
Strategy: Direct Mapping

Apple MLX

API: mlx.core.cumsum
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.ndarray.cumsum
Strategy: Direct Mapping