Tmp

a = np.array([[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]])
    
                       Axis 1
         ┌───────────────────────────▶
         │      0     1     2    3                   
         │    ┌─────┬─────┬────────────── 
         │  0 │ 10  │ 20  │ 30   40
         │    │ a0,0│ a0,1│ a0,2│ a0,3│
         │    ├─────┼─────┼────   │
         │  
  Axis 0 │ 1   50    60   70   80 
         │
         │  2   90    100  110  120
         │
         ▼