Модули PyTorch, работающие с изображениями, требуют от тензоров схемы измерений C × H × W (каналы, высота и ширина соответственно).