I am practicing on figuring out how to use the as_strided function in numpy. I began with the following example of my own where I generate 5 images of 3 x 3 where each image is full of 1s, the next full of 2s, and so on till 5. If I want convert the (5,3,3) volume to place all images into a single row, I can do the following which works: