The way the printing works is that we plot the last two dimensions as a 2D matrix.
And extra dimensions before are stacked and these 2D matrices are written one below the other.
Note that the same thing is done by numpy and most libraries I know.
In your case, because the last dimension is 1, then have only colum vectors.
You can remove that dimension if you want to change the way it is printed.