Det är alltid en lite blandad känsla att skriva ganska bra grundprincipkod för att göra någon tensoromläggning, bara för att upptäcka att PyTorch har en inbyggd funktion som gör det snabbare. Jag hade sett till att åtminstone skumma igenom dokumentationen för varje brännare och tensorfunktion, men om du inte vet vad du skulle använda något till, kommer det troligen inte fastna i huvudet när tillfälle ges. Pixel_unshuffle, tittar på dig.