megengine.functional.cumsum

cumsum(inp, axis)

计算给定轴上的向量元素的累积和。

参数:

• inp (Tensor) – 输入张量。 应该有一个数字数据类型。

• axis (int) – 指明需要累加的维度

返回:

需要进行累加操作的向量。

实际案例

如果:math:x_i`是``NaN`，则累加的合也是``NaN``(即., NaN 值传播).

实际案例

>>> x = Tensor([[1, 2, 3], [4, 5, 6]])
>>> F.cumsum(x, axis = 0)
Tensor([[1 2 3]
 [5 7 9]], dtype=int32, device=xpux:0)
>>> F.cumsum(x, axis = 1)
Tensor([[ 1  3  6]
 [ 4  9 15]], dtype=int32, device=xpux:0)