Piège
Convention features-en-lignes : sortie A2 a shape (10, N) (10 classes, N exemples). argmax(dim=0) retourne l'indice de classe pour chaque exemple — correct. argmax(dim=1) retourne, pour chaque classe, l'exemple le plus 'probable' — n'importe quoi. Sans dim, Pytorch fait l'argmax global (un seul entier sur tout le tenseur). Le notebook Cell 17 utilise dim=0 correctement.