标签: tensorflow-datasets
如果我使用
dataset.batch(n).prefetch(m),
批次或m个样本将被预取?
答案 0 :(得分:13)
Dataset.prefetch(m)转换预取其直接输入的m个元素。在这种情况下,由于其直接输入为dataset.batch(n),并且该数据集的每个元素都是一批(n元素),因此它将预取m 批次。
Dataset.prefetch(m)
m
dataset.batch(n)
n