将列表写入和读取到TFRecord示例

import tensorflow as tf

# 定义写入TFRecord的函数
def write_to_tfrecord(data_list, filename):
    writer = tf.io.TFRecordWriter(filename)
    for data in data_list:
        feature = {
            'data': tf.train.Feature(int64_list=tf.train.Int64List(value=data))
        }
        example = tf.train.Example(features=tf.train.Features(feature=feature))
        writer.write(example.SerializeToString())
    writer.close()

# 定义从TFRecord读取数据的函数
def read_from_tfrecord(filename):
    dataset = tf.data.TFRecordDataset(filename)
    feature_description = {
        'data': tf.io.FixedLenFeature([3], tf.int64)
    }
    def _parse_function(example_proto):
        return tf.io.parse_single_example(example_proto, feature_description)
    parsed_dataset = dataset.map(_parse_function)
    for data in parsed_dataset:
        print(data['data'].numpy())

# 示例数据
data_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
filename = 'data.tfrecord'

# 写入TFRecord文件
write_to_tfrecord(data_list, filename)

# 从TFRecord文件读取数据
read_from_tfrecord(filename)