for data_batch, labels_batch in train_generator: print("data_batch shape --> ",data_batch.shape) print("labels_batch shape --> ",labels_batch.shape) break
时间: 2023-03-28 17:03:02 浏览: 99
Tensorflow-Dataset-Tutorial:我的中级文章笔记本,有关如何在TensorFlow中使用Dataset API
这是一个关于 Python 代码的问题,data_batch 和 labels_batch 是训练数据的批次和标签的批次,通过 train_generator 生成器来获取。在循环中,打印出 data_batch 和 labels_batch 的形状,并使用 break 语句来跳出循环。
阅读全文