Tensorflow 使用数据集生成器进行tf.keras预测

我的tf.dataset是数据集生成器。使用它对tf.keras predict（）进行预测，我得到的输出比原始输入大数倍

我只是使用表单的生成器

 def gen():
    for ex in features:
        d = {k: v for k, v in asdict(ex).items() if v is not None}
        label = d.pop("label")
        yield (d, label)

input_names = ["input_ids"] + tokenizer.model_input_names

dataset=tf.data.Dataset.from_generator(
    gen,
    ({k: tf.int32 for k in input_names}, tf.int64),
    ({k: tf.TensorShape([None]) for k in input_names}, tf.TensorShape([])),)

没有重复或其他任何事情

这个生成器实际输出什么来做预测？如何控制尺寸。为什么输出大于字典的原始大小

---

如何正确使用生成器进行预测？

请将完整代码与示例数据共享以复制您的问题？请将完整代码与示例数据共享以复制您的问题？