神经网络中用到迭代器的作用

Question

test_db = tf.data.Dataset.from_tensor_slices((x_test, y_test))  # 构建测试集
# 预处理，批量化
test_db = test_db.map(preprocess).batch(512)
# 采样一个样本
sample = next(iter(train_db))

在预测中发现忘记加了第三句代码，好像也能训练出比较好的模型，想知道这句的作用，(是为了提取下一个train_db?但是，这里后面并没有用到sample?)