open3d.ml.torch.dataloaders.ConcatBatcher

class open3d.ml.torch.dataloaders.ConcatBatcher(device)

ConcatBatcher for KPConv

__init__(device)

Initialize

Parameters

device – torch device ‘gpu’ or ‘cpu’

Returns

The corresponding class.

Return type

class

collate_fn(batches)

collate_fn called by original PyTorch dataloader

Parameters

batches – a batch of data

Returns

the batched result

Return type

class