pytorch：“for”语句每次输出一个（1*n）的tensor，如何实现所有的tensor按位累加且取平均值

Question

第一次for循环输出tensor[1,2,3],
第二次for循环输出tensor[3,2,1]
希望能得到[1+3]/2,[2+2]/2,[3+1]/2,
最后是tensor[2,2,2]

Answer 1

转numpy结构，直接new = (numpy1+numpy2) / 2
保证shape和datatype完全一致就行