TensorFlow中组合训练数据函数tf.train.batch与tf.train.batch_join的输入维度与输出维度的关系
TensorFlow读出TFRecord中的数据,然后再经过预处理操作,此时需要注意:数据还是单个,而网络的输入一般以Batch为单位,因此我们需要将单个的数据组合成一个Batch,做为神经网络的输入。
TensorFlow提供组合训练数据的函数有四个:tf.train.batch(),tf.train.shuffle_batch()与tf.train.batch_join、tf.train.sh...
原创
2018-10-16 20:49:16 ·
1726 阅读 ·
1 评论