tf.data.Dataset + tf.lookup.index_table_from_file导致“表未初始化”错误

import tensorflow as tf

# 确保索引表文件存在并且路径正确
table_file = "path/to/table_file.txt"
assert tf.gfile.Exists(table_file), "Table file does not exist."

# 创建索引表
table = tf.lookup.index_table_from_file(table_file, num_oov_buckets=1)

# 初始化所有的表
init_op = tf.tables_initializer()

# 创建输入数据集
dataset = tf.data.Dataset.from_tensor_slices(["apple", "banana", "orange"])

# 对输入数据进行索引转换
dataset = dataset.map(lambda x: table.lookup(x))

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

# 初始化表
with tf.Session() as sess:
    sess.run(init_op)

    # 初始化迭代器
    sess.run(iterator.initializer)

    # 迭代获取数据
    while True:
        try:
            data = sess.run(iterator.get_next())
            print(data)
        except tf.errors.OutOfRangeError:
            break