Run
[ [2, 1], [5, 4] ] axis: 0 1 1 [2, 1] and [5, 4] are the elements of axis=0.
Output