Implement Transformer From Scratch Pytorch Version