https://adeveloperdiary.github.io/data-science/deep-learning/nlp/coding-transformer-model-from-scratch-using-pytorch-part-2/