wang3147

代码改成这样，跑出的图片跟作者的差距比较大，attn[...,skip_len:(skip_len +k_w **2),skip_len:(skip_len + q_w **2)]

attn[...，skip_len：（skip_len +k_w **2）]原代码好像就有问题

> 实际上我已经删除了这一行（它可能不是最好的解决方案），并且运行代码没有问题。希望我们能得到正确的解决方案 Hello, can you share your code

> 抱歉打扰您，因为我还是没有找到热力图的代码，不知道已经公布了吗？ 你好，我也刚好需要复现，请问你现在会了吗