Normalisation of attention scores

[soroorh]

For plots and comparisons in the notebooks, it will probably be more informative to normalise the scores to unit length, or simply divide by max value, whichever is appropriate.