def METHOD_NAME(self, input):
    input_id = input['input_ids'].unsqueeze(0)
    label = input_id[:, -1].to(self.device)
    pad_len = self.pad_len - input_id.shape[1]
    label_index = -2 - pad_len
    input_id = pad(input_id, (0, pad_len), value=1)
    return (input_id, label, label_index)