Question on Chapter 7.4.4 pooling layger output reshape

[Hustwireless]

I've got a question on the following line of code. According to the illustration of the book, I think this line

deep-learning-from-scratch/common/layers.py

Line 266 in cb755e9

out = out.reshape(N, out_h, out_w, C).transpose(0, 3, 1, 2)

perhaps should be

out = out.reshape(N, C, out_h, out_w)

because the tensor here seems to have a different struture from the convolution computation. However, I'm not quite sure about this, please do not hesitate to correct me if I'm wrong.

Thanks in advance!