about the attention map

[yangfan293]

Hello and thanks for your great work! I want to visualize the attention diagram in the dtr, and I would like to ask if the value of attention is the output of the "Z" in the following code?