Compute a Gradient with PyTorch AutogradDeep-ML | Problem