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

>>阅读原文<<