在中指定张量形状时
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)) ) ...
