Sorry for bothering. I can't understand where is the [batch['pad_mask']](https://github.com/dvl-tum/center-group/blob/9e55c06bbc9c59f7eb3d86c847f9a7880c164165/centergroup/models/centergroup.py#L243) from? Is its definition from [results['pad_mask']](https://github.com/dvl-tum/center-group/blob/9e55c06bbc9c59f7eb3d86c847f9a7880c164165/centergroup/data/transforms.py#L222)? or, does it have something to do with the 'mask' in the following image? 