torch的cumsum
函数作用:在张量的某个维度上进行累加从语言描述上比较抽象,看下以下代码加以说明(针对二维张量):import numpy as npx = 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