小编Yik*_* Ng的帖子

dataset = dataset.batch(50)
dataset = dataset.prefetch(buffer_size=1)

是预取1个批次还是1个元素？

根据tensorflow中的API文档，buffer_size是预取元素的最大数量。但似乎是在批处理数据集后进行了批处理。