Visualize: Figure 2
Description: Attention Distribution