Convolutional Variational Autoencoder code