Solved by tom in post #3 
              
              
                Yes, but you haven’t changed pred_R_imgs but only it’s clones. 
What you likely want is something like 
pred_R_imgs = torch.where(pred_R_imgs.abs()>1e-4 , pred_R_imgs, torch.tensor(1e-4))

Best regards 
Thomas