https://github.com/SmallMunich/nutonomy_pointpillars/blob/master/second/pytorch/models/pointpillars.py