Variational autoencoder in TensorFlow