Imported from https://github.com/nghuyong/ERNIE-Pytorch