torch的cumsum

函数作用：在张量的某个维度上进行累加
从语言描述上比较抽象，看下以下代码加以说明（针对二维张量）：

import numpy as np

x = np.array([[1, 2, 3], [4, 5, 6]])
x = torch.from_numpy(x)

这时x的结果为：

tensor([[1, 2, 3],
        [4, 5, 6]])

使用cumsum：

x_ = x.cumsum(dim=0)

x_的结果为：

tensor([[1, 2, 3],
        [5, 7, 9]])

当dim=0时表示在行的维度上进行累加，dim=1表示在列的维度上进行累加：

x_ = x.cumsum(dim=1)

x_的结果为：

tensor([[ 1,  3,  6],
        [ 4,  9, 15]])