def batch_xc(batch: dict, i: int):
    """Get the context inputs for a particular output dimension.

    Args:
        batch (dict): Batch dictionary.
        i (int): Index of output.

    Returns:
        tensor: Context inputs.
    """
    return batch["contexts"][i][0]