def METHOD_NAME(self, input_dict: Dict[str, Any]):
    r"""Collect the input ids for the given examples.
    Args:
        input_dict: A dict containing hyp/src/ref sentences.
    Returns:
        The token ids for each example.
    """
    output_sents = [
        self.tokenizer(
            input_dict['hyp'], return_tensors='pt',
            padding=True)['input_ids']
    ]
    if self.input_format == InputFormat.SRC or self.input_format == InputFormat.SRC_REF:
        output_sents += [
            self.tokenizer(
                input_dict['src'], return_tensors='pt',
                padding=True)['input_ids']
        ]
    if self.input_format == InputFormat.REF or self.input_format == InputFormat.SRC_REF:
        output_sents += [
            self.tokenizer(
                input_dict['ref'], return_tensors='pt',
                padding=True)['input_ids']
        ]
    input_ids = combine_input_sentences(output_sents, self.max_len,
                                        self.pad_token_id,
                                        self.eos_token_id)
    return input_ids