torch_ecg.utils.default_collate_fn#

torch_ecg.utils.default_collate_fn(batch: Sequence[Tuple[ndarray, ...] | Dict[str, ndarray]]) → Tuple[Tensor, ...] | Dict[str, Tensor][source]#

Default collate functions for model training.

The data generator (Dataset) should generate (__getitem__) n-tuples signals, labels, ..., or dictionaries of tensors.

Parameters:: batch (sequence) – Sequence of n-tuples, in which the first element is the signal, the second is the label, …; or sequence of dictionaries of tensors.
Returns:: Tuple or dict of Tensor, which are the concatenated values to feed into neural networks.
Return type:: tuple or dict