TAG Variational autoencoder