def select(data: dict, index: int=0):
    """ Select datum from batched dict values.

    :param data: Dictionary to batched values.
    :param index: Batch index.
    :return: Dictionary without batch dim.
    """
    selected = {}

    for key, value in data.items():
        selected[key] = value[index]

    return selected