Hi, I am not looking for something like this
You can save attention weights, then use matplotlib visualize.