Hi Frank,
I Figured there wouldn’t straightforward way to do it unfortunately 
I am doing differentiable rendering and thought that perhaps using this float tensor and casting it as an int when using it arange would still let me optimize over it 
I am trying to find an other way, perhaps using some sparse tensor or having a very high number of points and learn some float array dropping part of it, but haven’t found anyway yet.
Justin