zero.collate

zero.collate(iterable)

Almost an alias for torch.utils.data.dataloader.default_collate.

Namely, the input is allowed to be any kind of iterable, not only a list. Firstly, if it is not a list, it is transformed to a list. Then, the list is passed to the original function and the result is returned as is.

Parameters

iterable (Iterable[T]) –

Return type

Any