tensorflow 训练过程,使用tf.data迭代数据