Solved by Niv_Nayman in post #12 
              
              
                The issue is solved, with both the kind help from @JuanFMontesinos and at https://github.com/kornia/kornia/issues/682 
 [learn_x_translation_0.30000001192092896_y_translation_-0.20000000298023224_entropy_0.01_step_] 
The  only  problem was placing  translations = torch.stack([x_translation, y_transl…