pimm.datasets.utils.collate_fn#

collate_fn(batch, mix_prob=0)[source]#

Recursively collate tensors, mappings, and list-style point samples.

Tensor leaves are concatenated instead of stacked. Keys containing offset are treated as per-sample lengths and converted into cumulative offsets after concatenation.