Collator#

class Collator[source]#

Used for merging a list of samples to form a mini-batch of Tensor(s). Used when using batched loading from a dataset. Modified from pytorch/pytorch