tf.dataset.prefetch（）buffer_size的含义

Question

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

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

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

Answer 1

由于您使用dataset.prefetch(buffer_size=1)after dataset.batch()，这意味着它将预取 1 个批次。