def METHOD_NAME(batch_examples):
    if padding:
        input_ids = pad_func([example for example in batch_examples])
    else:
        input_ids = np.asarray([example for example in batch_examples], dtype=np.int64)
    return input_ids