I am doing a natural language generation task and following this tutorial - https://www.kdnuggets.com/2020/07/pytorch-lstm-text-generation-tutorial.html
In the LSTM I