tensorflow tfrecord 读写 实例

import tensorflow as tf
tf_file_writer = tf.python_io.TFRecordWriter("tmp")
fea = [1,2,3]#tf.ones(shape=[3,2],dtype=tf.float32)
label_vector= [4,5,6]#tf.zeros([2],dtype=tf.float32)
example = tf.train.Example(features=tf.train.Features(feature={
    'fea': tf.train.Feature(float_list=tf.train.FloatList(value=fea)),
    'fea_shape': tf.train.Feature(int64_list=tf.train.Int64List(value=[1,1,1])),
    'label': tf.train.Feature(float_list=tf.train.FloatList(value=label_vector))
}))
fea2 = [10,20,30]
label_vector2= [40,50,60]
example2 = tf.train.Example(features=tf.train.Features(feature={
    'fea': tf.train.Feature(float_list=tf.train.FloatList(value=fea2)),
    'fea_shape': tf.train.Feature(int64_list=tf.train.Int64List(value=[1,1,1])),
    'label': tf.train.Feature(float_list=tf.train.FloatList(value=label_vector2))
}))