使用string_input_producer从TFRecord文件读取数据时出错

import tensorflow as tf

# 创建一个TFRecordDataset对象
dataset = tf.data.TFRecordDataset(['path/to/file1.tfrecord', 'path/to/file2.tfrecord'])

# 定义解析函数
def parse_fn(example_proto):
    # 定义解析规则
    feature_description = {
        'feature1': tf.io.FixedLenFeature([], tf.int64),
        'feature2': tf.io.FixedLenFeature([], tf.string),
        # 添加其他特征...
    }
    # 解析单个样本
    parsed_example = tf.io.parse_single_example(example_proto, feature_description)
    return parsed_example

# 对数据集应用解析函数
dataset = dataset.map(parse_fn)

# 创建迭代器
iterator = dataset.make_one_shot_iterator()

# 获取下一个样本
next_example = iterator.get_next()

# 创建会话并运行
with tf.Session() as sess:
    try:
        while True:
            example = sess.run(next_example)
            # 处理样本数据
            # ...
    except tf.errors.OutOfRangeError:
        pass