https://github.com/MorvanZhou/PyTorch-Tutorial/blob/51c6c660c7658c13c2db9107a1cc54dadf240bd4/tutorial-contents/402_RNN_classifier.py#L74