Завжди трохи суперечливо, коли пишеш досить хороший код першого принципу для тензорної перестановки, а потім виявляєш, що в PyTorch є вбудована функція, яка робить це швидше. Я спеціально переглядав документи кожного факела та тензорної функції, але якщо ви не знаєте, для чого щось використати, це, ймовірно, не запам'ятається, коли з'явиться така можливість. Pixel_unshuffle, дивлюся на тебе.