C'est toujours un sentiment légèrement mitigé d'écrire un code basé sur des principes fondamentaux assez bon pour faire un réarrangement de tenseurs, seulement pour découvrir que PyTorch a une fonction intégrée qui le fait plus rapidement. J'avais pris soin de parcourir au moins la documentation de chaque fonction torch et tenseur, mais si vous ne savez pas à quoi vous servirait quelque chose, cela ne restera probablement pas en tête pour quand l'occasion se présentera. Pixel_unshuffle, je te regarde.