Cette solution sous-entend qu'on avait déjà un tenseur sur 3 dimensions. En réalité les deux dimensions spatiales de l'image étaient mélangées dans une même dimension (les lignes sont mises les unes à la suite des autres) donc le reshape permet de recréer cette forme 2D. Pour l'ajout de la dernière dimension, oui vous avez raison, indexer par None permet de faire exactement pareil.