pimm.models.polarmae.data.packed_to_batched#
- packed_to_batched(feat: Tensor, offset: Tensor)[source]#
Convert pimm packed format to PoLAr-MAE padded/batched format.
- Parameters:
feat – (N_total, C) packed features (e.g. [x, y, z, energy]).
offset – (B,) cumulative sum of point counts per sample.
- Returns:
points – (B, N_max, C) zero-padded tensor. lengths: (B,) number of valid points per sample.