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