Merge pull request #96 from innerlee/zz/dim

Fix: dim should be int

torch_scatter/utils/gen.py

@@ -8,7 +8,8 @@ import torch
 def maybe_dim_size(index, dim_size=None):
     if dim_size is not None:
         return dim_size
-    return index.max().item() + 1 if index.numel() > 0 else 0
+    dim = index.max().item() + 1 if index.numel() > 0 else 0
+    return int(dim)
 
 
 def broadcast(src, index, dim):