Regarding the figures in the article

[camallia1]

![0a44f215308c893ddd6ebc228675cd1]
Hello, I am a beginner. May I ask how the heat map about attention mechanism in the article is drawn, and how can I create variables for drawing images?Could you please give me an example？

[WenWeiTHU]

It is the attention scores from the trained model by passing an input multivariate series and then visualized, which is formulated as follows:

[camallia1]

Thank you for your answer. How should I call the A matrix in FullAttention? Could you give me a code example

[WenWeiTHU]

https://github.com/thuml/iTransformer/blob/main/layers/SelfAttention_Family.py#L147