Figure–Equation Mismatch

[httang1224]

The equation above appears to be inconsistent with Figure 3(b). According to the descriptions in the appendix and the implementation code, the attention mask seems to be applied to the target latent variable z_t, rather than to the reference latent.