求助一些论文的注意力模块的图是怎么画的

---

就一些文章中经过注意力模块后的热力图是如何画的，有没有比较好的code，或者方法，求助，自己不会画啊。

@Lay 我以前用过matplolib里面的一个方法。

plt.figure(figsize=(8, 6))
plt.subplots_adjust(left=.2, right=0.95, bottom=0.15, top=0.95)
#设置标注前后左右的距离
plt.imshow(scores, interpolation='nearest', cmap=plt.cm.hot,
        norm=MidpointNormalize(vmin=0.2, midpoint=0.92))
plt.xlabel('gamma')
plt.ylabel('C')
plt.colorbar()
plt.xticks(np.arange(len(gamma_range)), gamma_range, rotation=45)
plt.yticks(np.arange(len(C_range)), C_range)
plt.title('Validation accuracy')
plt.show()

仅供参考

有没有更加具体的方法啊

这个热力图是一个很经典的问题，貌似matplotlib里面有函数。