cumulative_sum(x, axis=0, *, name=None)¶
Calculates the cumulative sum of a tensor along a specified dimension.
x (np.ndarray or Tensor) – The tensor whose elements you want to sum. It must have at least one dimension.
axis (int, optional) – The dimension along which you want to sum the tensor. Defaults to 0.
name (str, optional) – The name of the node.
The cumulative sum of x along dimension axis.
- Return type