About Attention Visualization

[zzxxxfgb]

Liu, thank you for your work. I am very interested in the attention visualization in the paper. I tried to implement this in my project, but encountered some problems. Can you provide the relevant code? Thanks