collate_tensor

lightkit.data.collate_tensor(batch)[source]

Collates a tuple of batch items into the first tensor.

Might be useful if only a single tensor is passed to DataLoader.

Return type

Tensor