Tensorflow 数据集 API: dataset.batch(n).prefetch(m) 预取 m 个批次或样本?
Tensorflow Dataset API: dataset.batch(n).prefetch(m) prefetches m batches or samples?
如果我使用
dataset.batch(n).prefetch(m),
m 个批次或 m 个样本将被预取?
Dataset.prefetch(m)
转换预取其直接输入的 m
元素。在这种情况下,由于它的直接输入是 dataset.batch(n)
并且该数据集的每个元素都是一个批次(n
个元素),它将预取 m
batches.
如果我使用
dataset.batch(n).prefetch(m),
m 个批次或 m 个样本将被预取?
Dataset.prefetch(m)
转换预取其直接输入的 m
元素。在这种情况下,由于它的直接输入是 dataset.batch(n)
并且该数据集的每个元素都是一个批次(n
个元素),它将预取 m
batches.