Visualization code wanted

[lmz123321]

Hi, Thanks for your great work!
In your paper, attention maps of particular query points are shown. Could you share this visualization code?
More specifically, did you implement an interactive interface to do the visualization?

Many thanks again.

[xvjiarui]

Please find the visualization code here.