better import

torch_scatter/__init__.py

-from .functions import (scatter_add_, scatter_add, scatter_sub, scatter_sub_,
-                        scatter_mul, scatter_mul_, scatter_div, scatter_div_)
-
-__all__ = [
-    'scatter_add_', 'scatter_add', 'scatter_sub_', 'scatter_sub',
-    'scatter_mul_', 'scatter_mul', 'scatter_div_', 'scatter_div'
-]
+from .functions import *  # noqa

torch_scatter/functions/__init__.py

@@ -36,3 +36,9 @@ def scatter_div_(output, index, input, dim=0):
 def scatter_div(index, input, dim=0, max_index=None, fill_value=1):
     output = gen_output(index, input, dim, max_index, fill_value)
     return scatter_div_(output, index, input, dim)
+
+
+__all__ = [
+    'scatter_add_', 'scatter_add', 'scatter_sub_', 'scatter_sub',
+    'scatter_mul_', 'scatter_mul', 'scatter_div_', 'scatter_div'
+]