Imported from https://github.com/jacobgil/pytorch-grad-cam