Imported from https://github.com/pingpong-ai/xlnet-pytorch