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列累加到其他列；

>>阅读原文<<