in src/model_def.py [0:0]
def sampler(latent_dim, x): z_mean = layers.Dense(latent_dim)(x) z_log_var = layers.Dense(latent_dim)(x) z = layers.Lambda(sample_z)([latent_dim, z_mean, z_log_var]) return z_mean, z_log_var, z