GitHub repository: JasonLeeFdu/stnbhwd
Description: Modules for spatial transformer networks (BHWD layout)