Great answer and great examples! I have a question: can you explain why the m_reshape gives us the result we get? The original dimensions are
B x N x H x W
Now,  when we reshape, we want something like:
B x H x W x N , so that the images are side-by-side/concatenated together along the x-axis. Here, N = 2, so we should have two images.
So why do we get 4 images with the reshape? Why don’t we get something like the top half of the dog on the left, then the bottom half of the dog on the right? Thanks!