I am working on a small project involving a Spatial Transformer Network (STN) to process images. I accidentally uploaded a branch with untested code, and now I’m facing an issue where my image tensor output is all zeros.