Imported from https://github.com/alexis-jacq/Pytorch-Sketch-RNN