Attention visualization code: Fig. 4 from Appendix A.3

[vadimkantorov]

Hi!

Would you have any guidance / code on reproducing the decoder attention visualization? (fig. 4 from A.3). I'm worried of making some padding-related mistakes while working with sampled_locations

Thanks!