Codes for visualizing attention weights just like in the CTX paper Fig.3

[buncybunny]

I'm trying to visualize the softmax normalized attention weights ā as the Figure 3 in the CTX paper.
I couldn't find the proper codes for this visualizing task anywhere in the repo. Can you please provide codes for this?
Or at least give me some guideline for implementing it?

[Zhou-Yuen]

Hello,I wonder if you have get the code of visualization

[buncybunny]

@Zhou-Yuen Hey sorry for being late. Unfortunately I couldn't get the visualization codes.

[Zhou-Yuen]

It doesn't matter. Thank you very much for your reply

…

------------------ 原始邮件 ------------------ 发件人: ***@***.***>; 发送时间: 2022年1月5日(星期三) 下午4:02 收件人: "google-research/meta-dataset"; 抄送: "Mention"; 主题: Re: [google-research/meta-dataset] Codes for visualizing attention weights just like in the CTX paper Fig.3 (#74) @Zhou-Yuen Hey sorry for being late. Unfortunately I couldn't get the visualization codes. — Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android. You are receiving this because you were mentioned.Message ID: ***@***.***>