voxelize bug?

[victoryc]

The intention seems to be that "coords" passed to voxelize_idx function in voxelize.cpp may or may not include the batch indices. The assert "assert(coords.size(1) >= dimension and coords.size(1) <= dimension + 1" confirms that intention. But, in voxelize_outputmap, there is the following code:
LongInt *coord = coords + inputIdx * (dimension + 1);
This assumes that batch index is always present.
The way I see it, it is better to pass coords.size(1) as a parameter to voxelize_output function and use that instead of (dimension + 1).