fix

test/dense_grid.json

@@ -43,8 +43,8 @@
     "position": [[0, 0], [11, 9], [2, 8], [2, 2], [8, 3], [1, 1], [6, 6]],
     "size": [5, 5],
     "batch": [0, 0, 0, 0, 0, 1, 1],
-    "start": [0],
-    "end": [20],
+    "start": [0, 0],
+    "end": [20, 20],
     "expected": [0, 9, 1, 0, 4, 16, 21],
     "expected_C": 32
   }

torch_cluster/functions/grid.py

@@ -51,9 +51,6 @@ def _fixed_cluster_size(position, size, batch=None, end=None):
     if end is None:
         return _minimal_cluster_size(position, size)
 
-    assert end.numel() == size.numel(), (
-        'End tensor must have same size as size tensor')
-
     end = end.type_as(size)
     eps = 0.000001  # Simulate [start, end) interval.
     if batch is None: