a = np.random.randint(0,100,size=(2,3,4,5)) a
对某一维度求和,可以理解为用和消除该维度,如对维度0求和,求和结果的维度是(3,4,5),那么就是用维度0对应位置的和(如第一个值32+98)来消除维度0的第一个值,其他值依次类推。如果是对维度1求和,就是用32+70+98消除维度1的第一个值,其他值以此类推。
python numpy 多维度求和
原文:https://www.cnblogs.com/learning-logs/p/12922069.html