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)之后使用dataset.batch()，这意味着它将预取1批。