Всегда немного смешанные чувства, когда пишешь довольно хороший код с нуля для перестановки тензоров, только чтобы обнаружить, что в PyTorch есть встроенная функция, которая делает это быстрее. Я старался хотя бы просмотреть документацию по каждой функции torch и tensor, но если ты не знаешь, для чего тебе что-то нужно, это, вероятно, не запомнится, когда появится возможность. Pixel_unshuffle, я на тебя смотрю.