Я новичок в pytorch, для одного из моих проектов я делю большое изображение на меньшие плитки / патчи.Я использую раскрытие, чтобы это произошло.Мой код выглядит следующим образом:
data = training_set[1][0].data.unfold(1, 64, 64).unfold(2, 64, 64).unfold(3, 64, 64)
После этого я транспонирую результирующую матрицу, так как изображения переворачиваются, как в этом примере кода
torch.t(data [0][0][0][0])
, но результирующие изображения теряют цвет или получаютпо какой-то причине обесцвечен, и я беспокоюсь, что это может повлиять на любые вычисления, которые я делаю на основе этих исправлений.
Ниже приведен скриншот проблемы 
Верх - это патч, а нижний - полная картинка
Любая помощь приветствуется, спасибо