self.batch_size = batch_size self.batches = batches
时间: 2024-04-19 09:03:50 浏览: 143
tensorflow 中dataset.padded_batch函数的个人理解过程
这段代码的作用是将输入数据集的批次大小 batch_size 和数据集 batches 存储到迭代器的属性中。self.batch_size 记录每个 batch 中数据的数量,self.batches 记录整个数据集中的所有数据点。这个迭代器可以用于将数据集划分成一个个大小为 batch_size 的 batch。
阅读全文