Pytorch code for spatial transformer network with mask - View it on GitHub