Merge branch 'master' of https://github.com/rusty1s/pytorch_cluster

a353f274 · rusty1s · eae34d5d · 7ce4c1ca · a353f274
Commit a353f274 authored Dec 19, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

cpu/grid.cpp cpu/grid.cpp +2 -2

No files found.
--- a/cpu/grid.cpp
+++ b/cpu/grid.cpp
@@ -8,8 +8,8 @@ at::Tensor grid(at::Tensor pos, at::Tensor size, at::Tensor start,
  num_voxels = num_voxels.cumprod(0);
  num_voxels = at::cat({at::ones(1, num_voxels.options()), num_voxels}, 0);
-  auto index = empty(size.size(0), num_voxels.options());
+  auto index = at::empty(size.size(0), num_voxels.options());
-  arange_out(index, size.size(0));
+  at::arange_out(index, size.size(0));
  num_voxels = num_voxels.index_select(0, index);
  auto cluster = (pos / size.view({1, -1})).toType(at::kLong);