Model layout from the plot_model utility in KERAS. The upper part represents the encoder, the left branch the decoder, and the right the custom layer gex (g). The output of encoder_last is the latent space . Concatenate is a dummy layer that concatenates the output of the two branches. The question marks indicate the unknown batch size, that is defined at runtime. To be compared with Fig. 2.