How to draw a heatmap for decoder?

[131413ljk]

Hello,

Thanks for your great work. Excuse me, but I'm interested in how to get a heat map of a decoder like Figure 4 mentioned in the paper. Can you provide a visual script? Thank you!