Attention Cues

https://d2l.ai/chapter_attention-mechanisms-and-transformers/attention-cues.html

Hello I’m having difficulty getting an image of the figure. show_heatmaps(attention_weights, xlabel=‘Keys’, ylabel=‘Queries’) doesn’t generate a figure as it should. I’m working in an editor and execute the scripts directly without the repl. I’m using tensorflow.