How does the visualization of Attention Weights organize the code?

[choiseonb1n]

The _MultiheadAttention class of PatchTST_backbone.py returns Attention_weight.
However, I'm not sure how to use it to visualize it.

How do I organize the code to visualize it like Figure 6 in the paper?

[111lzx111]

Thanks to the authors for their excellent time series forecasting work.
I am having the same problem of how to visualize the attention weights shown in Figure 6 in the paper.

[iganggang]

@choiseonb1n @111lzx111 @yuqinie98
i have the same question. i want to visualize an attention map in time series, such as figure 6.
if u have any news, please kick me. Thank you very much.