numpy sum mean axis

import torch t=torch.randn(2,2,2) print(t.sum(0)) print(t.sum(1)) print(t.sum(2)) 假设t=[ [ [a, b], [c,d] ], [ [e, f], [g, h] ] ] 分成3层。第0层即t。第1层有两个数组， t1= t[0]= [ [a, b], [c,d] ], t2=t0[1]= [ [e, f], [g

>>阅读原文<<