from typing import Tuple
import torch


def collate_tensor(batch: Tuple[torch.Tensor, ...]) -> torch.Tensor:
    """
    Collates a tuple of batch items into the first tensor. Might be useful if only a single tensor
    is passed to :class:`~lightkit.data.DataLoader`.
    """
    return batch[0]