如何为返回可变行数据的生成器构建tf.data.from_generator的output_signature

首先明确结论：用None表示第一维（行数可变）是完全正确的。

在TensorFlow的TensorSpec中，None专门用来标记动态可变的维度，刚好匹配你生成器返回的行数不固定的numpy.ndarray——不管每次yield的data有多少行，这个签名都能正确识别并兼容。

不过要注意你代码里的一个小疏漏：你描述的data列数是500，但代码里写的是(None, 512)，这里得改成(None, 500)，不然会和实际数据的维度不匹配，后续可能触发维度错误。

修正后的代码如下：

dataset = tf.data.Dataset.from_generator(
   datagen_row,
   output_signature=(
      tf.TensorSpec(shape=(None, 500), dtype=tf.float32, name=None),
      tf.TensorSpec(shape=(), dtype=tf.int64, name=None)
   )
)

内容的提问来源于stack exchange，提问作者Frank