批量大小可变的TensorFlow DataSet from_generator

批量大小可变的TensorFlow DataSet from_generator

在中指定张量形状时

from_generator

，您可以将其

None

用作元素来指定尺寸可变的尺寸。这样，您可以容纳不同大小的批次，尤其是“剩余”批次，它比您要求的批次大小小一些。所以你会用

def make_fashion_dset(file_name, batch_size, shuffle=False):    dgen = _make_fashion_generator_fn(file_name, batch_size)    features_shape = [None, 28, 28, 1]    labels_shape = [None, 10]    ds = tf.data.Dataset.from_generator(        dgen, (tf.float32, tf.uint8),        (tf.TensorShape(features_shape), tf.TensorShape(labels_shape))    )    ...