def sampler()

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