Modules for spatial transformer networks (BHWD layout) - View it on GitHub