Imported from https://github.com/nyu-dl/bert-gen