Pytorch Set Parameters Manually Calculate