Cuando dos ejes comparten tamaño

Apila una foto (512, 512, 3) tres veces y obtienes un lote (3, 512, 512, 3). Conviértelo a NCHW y obtienes (3, 3, 512, 512).

Ahora dos ejes tienen tamaño 3: uno cuenta ejemplos y otro cuenta colores. Una vez que dos ejes comparten tamaño, la forma no te puede decir cuál es cuál. Solo tu propio seguimiento puede.

Esta es la mitad más profunda de la sección. Reshape contra transpose es un error que puedes detectar. Este solo lo puedes prevenir.