可视化的网络注意力层

我想知道如何可视化 VGGT 的每一层子注意力。希望得到您的建议。