clean up [skip ci]

script/conda.sh

@@ -24,9 +24,3 @@ fi
 conda update --yes conda
 
 conda create --yes -n test python="${PYTHON_VERSION}"
-
-gcc --version
-
-if [ "${TRAVIS_OS_NAME}" = "linux" ]; then
-  conda install gcc_linux-64 --yes
-fi

torch_scatter/scatter.py

@@ -50,11 +50,9 @@ 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.div_(count)
-        # out.true_divide_(count)
+        out.true_divide_(count)
     else:
-        out.div_(count)
-        # out.floor_divide_(count)
+        out.floor_divide_(count)
     return out