Solved by tom in post #4 
              
              
                If you have tensors and expect to read it in one go, I think it is very reasonable to just store them as tensors. It is a speed thing, too. Going from read from disk + CPU augmentation to store in memory + GPU augmentation can give a formidable speedup. 
(I did this as live coding in a workshop la…