bugfix

setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 import build  # noqa
 
-__version__ = '0.2.0'
+__version__ = '0.2.1'
 url = 'https://github.com/rusty1s/pytorch_scatter'
 
 install_requires = ['cffi']

torch_scatter/__init__.py

@@ -6,7 +6,7 @@ from .functions.mean import scatter_mean_, scatter_mean
 from .functions.max import scatter_max_, scatter_max
 from .functions.min import scatter_min_, scatter_min
 
-__version__ = '0.2.0'
+__version__ = '0.2.1'
 
 __all__ = [
     'scatter_add_', 'scatter_add', 'scatter_sub_', 'scatter_sub',

torch_scatter/functions/sub.py

@@ -51,7 +51,7 @@ def scatter_sub_(output, index, input, dim=0):
        -2 -4 -4  0  0  0
        [torch.FloatTensor of size 2x6]
     """
-    return output.scatter_add_(dim, index, -input)
+    return output.scatter_add_(dim, index, -1 * input)
 
 
 def scatter_sub(index, input, dim=0, size=None, fill_value=0):