Make zero_grad work with pytorch 1.11

dlutils/pytorch/jacobian.py

 import torch
-from torch.autograd.gradcheck import zero_gradients
 
 
 def jacobian(inputs, output):
@@ -19,7 +18,7 @@ def jacobian(inputs, output):
         J = J.cuda()
 
     for i in range(num_classes):
-        zero_gradients(inputs)
+        inputs.zero_grad()
         grad_output.zero_()
         grad_output[:, i] = 1
         output.backward(grad_output, retain_graph=True)