running sum

Computes the running sum of the input tensor along a given axis. For example, the running sum of is .