一起来用tf.data API!(6)——批处理数据集元素
(一)前 言
(二)简单的批处理
(1)创建Dataset
(2)实现批处理
(三)使用填充批处理张量
(四)总 结
(一)前 言
在上一节中我们介绍了如何使用tf.data API读取TFRecords文件,在这一节中我们将介绍如何对数据集元素进行批处理。
(二)简单的批处理
(1)创建Dataset
以range函数为例:
(2)实现批处理
(三)使用填充批处理张量
上述方法适用于具有相同大小的张量。不过,很多模型(例如序列模型)处理的输入数据可能具有不同的大小(例如序列的长度不同)。为了解决这种情况,可以通过 Dataset.padded_batch() 转换来指定一个或多个会被填充的维度,从而批处理不同形状的张量。
(四)总 结
在这一节中,我们介绍了使用tf.data API对数据进行批处理的操作方法,在下一节中,我们将对tf.data API对训练工作的流程的操作进行介绍,有任何的问题可在评论区留言,我会尽快回复,谢谢支持!