Tensorflow数据集API:dataset.batch(n).prefetch(m)预取m个批次或样本?

时间:2018-04-07 11:49:46

标签: tensorflow-datasets

如果我使用

dataset.batch(n).prefetch(m), 

批次或m个样本将被预取?

1 个答案:

答案 0 :(得分:13)

Dataset.prefetch(m)转换预取其直接输入的m个元素。在这种情况下,由于其直接输入为dataset.batch(n),并且该数据集的每个元素都是一批(n元素),因此它将预取m 批次