Imported from https://github.com/sebastiani/pytorch-attention-augmented-convolution