Imported from https://github.com/tensorflow/magenta/tree/master/magenta/models/svg_vae