Please have a look at how the formula's are shown: https://www.learnpytorch.io/08_pytorch_paper_replicating/