how to store numpy arrays as tfrecord?
问题 I am trying to create a dataset in tfrecord format from numpy arrays. I am trying to store 2d and 3d coordinates. 2d coordinates are numpy array of shape (2,10) of type float64 3d coordinates are numpy array of shape (3,10) of type float64 this is my code: def _floats_feature(value): return tf.train.Feature(float_list=tf.train.FloatList(value=value)) train_filename = 'train.tfrecords' # address to save the TFRecords file writer = tf.python_io.TFRecordWriter(train_filename) for c in range(0