Get the encoder, e(z|x):
with tf.variable_scope('ezx', reuse=tf.AUTO_REUSE):
ezx = ezx_dist(net)