注意力图

谢谢您的工作让我收获很多，我想请问一下文中的注意力图即figure. 4是通过什么方式绘制的？我尝试使用gradcam对准vit的最后一层norm，但得到效果不太理想，希望您能给我一些启发！