Originally imgs is [0 … 255] and maps is [0 … 1]. When I load imgs to a tensor, they become [0 … 1]. I am loading cifar-10 images, so for example, I have the batch [24, 3, 32, 32] for imgs and another for maps. I tried to do i = img.permute(0,2,3,1) and plot lets say i[0], but it doesn’t seems to be right.