Один из вариантов: просто разделить значения пикселей на 255 (максимальное число, которое можно представить с помощью беззнакового 8-битного числа)
Модули PyTorch, работающие с изображениями, требуют от тензоров схемы измерений C × H × W (каналы, высота и ширина соответственно).
Хранилище всегда представляет собой одномерный массив