comment

torch_spline_conv/functions/utils.py

@@ -37,6 +37,7 @@ def spline_weighting_forward(x, weight, basis, weight_index):
 
 def spline_weighting_backward(grad_output, x, weight, basis, weight_index):
     grad_input = x.new(x.size(0), weight.size(1))
+    # grad_weight computation via `atomic_add` => Initialize with zeros.
     grad_weight = x.new(weight.size()).fill_(0)
     func = get_func('weighting_backward', x)
     func(grad_input, grad_weight, grad_output, x, weight, basis, weight_index)