torch.cumsum() 和 torch.cumprod()

  import torch as t a = t.arange(0, 6).view(2,3) print(a) a.cumsum(dim=0)   a = t.arange(0, 6).view(2,3) print(a) a.cumsum(dim=1)   对于二维输入a，dim=0(第1行不动，将第1行累加到其他行)；dim=1(进入最内层，转化成列处理。第1列不动，将第1列累加到其他列；

>>阅读原文<<