Imported from https://github.com/edouardpineau/Variational-Recurrent-Neural-Networks-for-Graph-Classification