注意力热图绘制

作者您好:
   非常感谢，您和您团队在这方面的工作，我个人对这方面也非常感兴趣。但是在绘制注意力热图中发生了一些问题，请问您是用什么方法绘制的，如果方便的话能否公开一下这方面的源码。非常感谢！！！