写出相当不错的第一原则代码来进行一些张量重排，总是有一种复杂的感觉，结果发现 PyTorch 有一个内置函数可以更快地完成这项工作。 我曾经至少浏览过每个 torch 和 tensor 函数的文档，但如果你不知道某个函数的用途，当机会来临时，它可能不会在你脑海中留下印象。 Pixel_unshuffle，我在看着你。