Solved by JuanFMontesinos in post #2 
              
              
                You can use expand method to the single image. It points to the same mem adress than the original image. 
https://pytorch.org/docs/stable/tensors.html#torch.Tensor.expand