Visualize attention

[micha-net]

Hi, I am interested in contributing to the project by visualization the attention.

I am currently writing my Master Thesis in AI about this topic and already build a proof of concept for this project - see images below.
The only issue is, that I would just be able to add the code to this repository after I handed in my Thesis in ~3 months.

Let me know if you are interested in this. Happy to exchange ideas.