Merge pull request #13 from AntoinePrv/deprecation_warning

Replace saved_variables to saved_tensors.

torch_scatter/div.py

@@ -18,7 +18,7 @@ class ScatterDiv(Function):
 
     @staticmethod
     def backward(ctx, grad_out):
-        out, src, index = ctx.saved_variables
+        out, src, index = ctx.saved_tensors
 
         grad_src = None
         if ctx.needs_input_grad[1]:

torch_scatter/max.py

@@ -19,7 +19,7 @@ class ScatterMax(Function):
 
     @staticmethod
     def backward(ctx, grad_out, grad_arg):
-        index, arg = ctx.saved_variables
+        index, arg = ctx.saved_tensors
 
         grad_src = None
         if ctx.needs_input_grad[1]:

torch_scatter/min.py

@@ -19,7 +19,7 @@ class ScatterMin(Function):
 
     @staticmethod
     def backward(ctx, grad_out, grad_arg):
-        index, arg = ctx.saved_variables
+        index, arg = ctx.saved_tensors
 
         grad_src = None
         if ctx.needs_input_grad[1]:

torch_scatter/mul.py

@@ -18,7 +18,7 @@ class ScatterMul(Function):
 
     @staticmethod
     def backward(ctx, grad_out):
-        out, src, index = ctx.saved_variables
+        out, src, index = ctx.saved_tensors
 
         grad_src = None
         if ctx.needs_input_grad[1]: