https://analyticsindiamag.com/recurrent-neural-network-in-pytorch-for-text-generation/