torchoutil.utils.data.collate module

class AdvancedCollateDict(
pad_values: Dict[str, Any] | None = None,
include_keys: str | Pattern | Iterable[str | Pattern] | None = None,
exclude_keys: str | Pattern | Iterable[str | Pattern] | None = None,
key_mode: Literal['intersect', 'same', 'union'] = 'same',
)[source]

Bases: object

Collate class to automatically convert to tensor, pad sequences and filter keys in a batch dict.

class CollateDict(
key_mode: Literal['intersect', 'same', 'union'] = 'same',
)[source]

Bases: object

Collate class to handle a batch dict.