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 个样本?

m batches or m samples will be prefetched?

推荐答案

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

The Dataset.prefetch(m) transformation prefetches m elements of its direct input. In this case, since its direct input is dataset.batch(n) and each element of that dataset is a batch (of n elements), it will prefetch m batches.

这篇关于Tensorflow 数据集 API:dataset.batch(n).prefetch(m) 预取 m 个批次或样本?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！