Tmp

a = np.array([[A, B, C, D], [E, F, G, H], [I, J, K, L]])
    
                       Axis 1
         ┌──────────────────────────────────▶
         │         0     1     2    3                   
         │      ┌─────┬─────┬─────┬─────┐
         │      │  A  │  B  │  C  │  D  │ 
         │ a0 0 │ a0,0│ a0,1│ a0,2│ a0,3│
         │      ├─────┼─────┼─────┼─────┤
         │      │  E  │  F  │  G  │  H  │ 
  Axis 0 │ a1 1 │ a1,0│ a1,1│ a1,2│ a1,3│ 
         │      ├─────┼─────┼─────┼─────┤
         │      │  I  │  J  │  K  │  L  │
         │ a2 2 │ a2,0│ a2,1│ a2,2│ a2,3│ 
         │      └─────┴─────┴─────┴─────┘
         │         
         ▼