how does the axis parameter from numpy work?,"e.shape == (3, 2, 2)"