一起来用tf.data API！（6）——批处理数据集元素

一起来用tf.data API！（6）——批处理数据集元素

（一）前 言
（二）简单的批处理
    （1）创建Dataset
    （2）实现批处理
（三）使用填充批处理张量
（四）总 结

（一）前 言

在上一节中我们介绍了如何使用tf.data API读取TFRecords文件，在这一节中我们将介绍如何对数据集元素进行批处理。

（二）简单的批处理

（1）创建Dataset

以range函数为例：

（2）实现批处理

（三）使用填充批处理张量

上述方法适用于具有相同大小的张量。不过，很多模型（例如序列模型）处理的输入数据可能具有不同的大小（例如序列的长度不同）。为了解决这种情况，可以通过 Dataset.padded_batch() 转换来指定一个或多个会被填充的维度，从而批处理不同形状的张量。

（四）总 结

在这一节中，我们介绍了使用tf.data API对数据进行批处理的操作方法，在下一节中，我们将对tf.data API对训练工作的流程的操作进行介绍，有任何的问题可在评论区留言，我会尽快回复，谢谢支持！