寫出相當不錯的第一原則代碼來進行一些張量重排,總是有一種稍微混合的感覺,結果卻發現 PyTorch 有一個內建函數可以更快地做到這一點。 我曾經特意至少瀏覽過每個 torch 和 tensor 函數的文檔,但如果你不知道某個功能的用途,那麼當機會來臨時,它可能不會在你腦海中留下印象。 Pixel_unshuffle,我在看著你。