Merge pull request #13 from AntoinePrv/deprecation_warning

Replace saved_variables to saved_tensors.

Author: rusty1s

torch_scatter/div.py

@@ -18,7 +18,7 @@ def forward(ctx, out, src, index, dim):
 
     @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 @@ def forward(ctx, out, src, index, dim):
 
     @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 @@ def forward(ctx, out, src, index, dim):
 
     @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 @@ def forward(ctx, out, src, index, dim):
 
     @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]: