image/svg+xml Axis(0) Axis(1) Axis(2) .axis_iter(Axis(2)) Input shape: (3, 4, 5)Output shapes: (3, 4) 0 1 2 3 4