How to output a shape of (N,1) ?

[eyegotthis]

Hi there, I really enjoyed the read in the paper and appreciate a lot that you uploaded code for you work. One thing I do not get is how to get an output shape from the model in (N,1) where N being the total number of points in point cloud and 1 being the label for that specific point. Your network only outputs in the cylindrical voxel space if im not wrong and i would want it to be in the shape mentioned for semantickitti dataset. Would be glad if you could guide me here, Thanks!