Transformer Implementation Using Pytorch