Pytorch Weighted Gpa