import torch


def run_single(model: torch.nn.Module, *args) -> torch.Tensor:
    """ Runs a single element (no batch dimension) through a PyTorch model """
    return model(*[a.unsqueeze(0) for a in args]).squeeze(0)