Python tensorflow创建具有多个数组功能的tfrecord

import tensorflow as tf

# 定义第一个数组特征
feature1 = tf.train.Feature(float_list=tf.train.FloatList(value=[1.0, 2.0, 3.0]))

# 定义第二个数组特征
feature2 = tf.train.Feature(int64_list=tf.train.Int64List(value=[4, 5, 6]))

# 创建特征列表
feature_list = {
    'array1': feature1,
    'array2': feature2
}

# 创建tfrecord文件
writer = tf.io.TFRecordWriter('data.tfrecord')

# 创建Example对象
example = tf.train.Example(features=tf.train.Features(feature=feature_list))

# 将Example对象序列化并写入tfrecord文件
writer.write(example.SerializeToString())

# 关闭写入器
writer.close()