def sample_f(mean, logvar):
return seq2seq.sample(
mean,
logvar,
latent_dim,
iaf,
kl_min,
anneal,
self.kl_rate,
dtype)
def sample_f(mean, logvar):
return seq2seq.sample(mean,
logvar,
batch_size=batch_size,
latent_dim=latent_dim,
dtype=dtype)