Attention area when CNN

Now there are:

  • Attention map (the 2D matrix of important value)
  • Associated original image

Then I want to implement CNN, which pays more attention to the location of original image on the basis of attention map.

How can I achieve this idea in PyTorch?

May be you can check the ICLRV2018 paper “Learn to Pay attention”. It works on the same idea.