import torch


def add_dim(tensor: torch.Tensor, index: int,
            *args, **kwargs) -> torch.Tensor:
    """
    A wrapper function for torch.squeeze

    Args:
        tensor (): the tensor to unsqueeze
        index (): the index of to unsqueeze

    Returns:
        The unsqueeze tensor
    """
    return tensor.unsqueeze(index)