Imported from https://github.com/jaywonchung/BERT4Rec-VAE-Pytorch