我想要可视化最后一层的block的形状,但是好像最终输出的大小是B,C,,H,W大小,而注意力图是B,C,N大小,请问该如何u解决这个问题
我想要可视化最后一层的block的形状,但是好像最终输出的大小是B,C,,H,W大小,而注意力图是B,C,N大小,请问该如何u解决这个问题