Imported from https://github.com/leaderj1001/Attention-Augmented-Conv2d