I noticed that you drew an attention map in the text. How do you draw an attention map? Can you provide the source code?