adding Mask Autoregressive Flow to Variational Autoencoder TF.2-0

Question

I\'m studying normalizing Flow ,and I want to add some mask autoregressive flow layer to my VAE ,but I did not understand the procedure of how to do it . Did I need to add