import numpy as np
arr = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])
print(np.sum(arr,axis=0)) # [[6, 8], [10, 12]]
print(np.sum(arr,axis=1)) # [[4, 6], [12, 14]]
print(np.sum(arr,axis=2)) # [[3, 7], [11. 15]]
import numpy as np
arr = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])
print(np.sum(arr,axis=0)) # [[6, 8], [10, 12]]
print(np.sum(arr,axis=1)) # [[4, 6], [12, 14]]
print(np.sum(arr,axis=2)) # [[3, 7], [11. 15]]