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.