制作VQAencoder的注意力图

[JXH-2020]

您好，我这边在制作VQA注意力图的时候出现了问题，利用unify_multihead_attention.py中的attn_weights来画图，展现不了注意力的能力，如下图：

原图是：

请问该如何获取VQA的注意力图，我思考了很久，尝试了不少方法，但得不到解决，希望您能抽空帮忙看看解决一下，提供教程。