Can you give me the temporal attention map visualization code as shown in fig2? 