Piège
Pour un batch de N images 28×28, .flatten() donné un vecteur de N×784 éléments — mauvaise forme. Il faut préserver la dimension batch : .view(N, 28*28) ou .reshape(N, 784). Conséquence du mauvais reshape : le réseau prend N×784 entrées au lieu de N exemples de 784 features → erreur de dimension fatale ou silencieuse.