Error: la forma demuestra que el reshape funcionó

«Ambos resultados tienen forma (3, 2, 2), así que ambos son imágenes CHW válidas.»

hwc = np.arange(12).reshape(2, 2, 3)
right = hwc.transpose(2, 0, 1)
wrong = hwc.reshape(3, 2, 2)
assert right.shape == wrong.shape
assert right[1, 0, 0] == hwc[0, 0, 1] == 1
assert wrong[1, 0, 0] == 4

Las comprobaciones de forma pasan; una comprobación del valor del canal falla. right[1, 0, 0] es 1, el valor que de verdad corresponde ahí; wrong[1, 0, 0] es 4. Usa transpose para reordenar ejes.