clean up [skip ci]

README.md

@@ -40,10 +40,10 @@ All included operations are broadcastable, work on varying data types, are imple
 
 ### Binaries
 
-#### PyTorch 1.5.0
-
 We provide pip wheels for all major OS/PyTorch/CUDA combinations, see [here](https://pytorch-geometric.com/whl).
 
+#### PyTorch 1.5.0
+
 To install the binaries for PyTorch 1.5.0, simply run
 
 ```

torch_scatter/__init__.py

@@ -5,7 +5,6 @@ import os.path as osp
 import torch
 
 __version__ = '2.0.4'
-expected_torch_version = (1, 4)
 
 try:
     for library in ['_version', '_scatter', '_segment_csr', '_segment_coo']:

torch_scatter/scatter.py

@@ -50,9 +50,11 @@ def scatter_mean(src: torch.Tensor, index: torch.Tensor, dim: int = -1,
     count.clamp_(1)
     count = broadcast(count, out, dim)
     if torch.is_floating_point(out):
-        out.true_divide_(count)
+        out.div_(count)
+        # out.true_divide_(count)
     else:
-        out.floor_divide_(count)
+        out.div_(count)
+        # out.floor_divide_(count)
     return out