Fix: dim should be int

[innerlee]

Otherwise it may error:

     52         out_size[dim] = dim_size
     53         print(dim, dim_size)
---> 54         out = src.new_full(out_size, fill_value)
     55 
     56     return src, out, index, dim

TypeError: new_full(): argument 'size' must be tuple of ints, but found element of type float at pos 2

[codecov-io]

Codecov Report

Merging #96 into master will not change coverage.
The diff coverage is 100%.

@@          Coverage Diff          @@
##           master    #96   +/-   ##
=====================================
  Coverage     100%   100%           
=====================================
  Files          15     15           
  Lines         142    143    +1     
=====================================
+ Hits          142    143    +1

Impacted Files	Coverage Δ
torch_scatter/utils/gen.py	100% <100%> (ø)	⬆️

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 48024c1...41955ed. Read the comment docs.

[rusty1s]

Thanks :)