When the number of IAF is non zero, the latent dimension changes to the same value as the batch size.
When the number of IAF is non zero, the latent dimension changes to the same value as the batch size.