Tensor.cumsum()

Returns the cumulative sum of the elements along a given axis.

Usage

from tinygrad.tensor import Tensor
 
a1 = Tensor([[1, 2, 3]])
a2 = Tensor([[1, 2, 3]])
 
a1 = a1.cumsum(1)
 
print(a1.numpy())

[[1. 3. 6.]]

;