How can I visualize the attention maps in Figure 6? Can you provide the corresponding visualization code?

[jlqzzz]

How can I visualize the attention maps in Figure 6?

Can you provide the corresponding visualization code?

Thank you very much！