Solved by pfloat in post #7 
              
              
                Hello, 
ok then it means, it is possible to group the float values by 3 to get color pixels assuming the floats are gathered in a list and that this is the original intended format for the file. 
Now a solution would be to subclass the torch.utils.data.Dataset class: 
import itertools
from torch.uti…