tensorflow保存/恢复队列

import tensorflow as tf

# 创建一个队列
queue = tf.FIFOQueue(100, tf.float32)
enqueue_op = queue.enqueue([1.0, 2.0, 3.0])

# 创建一个保存器
saver = tf.train.Saver(var_list=tf.global_variables())

with tf.Session() as sess:
    # 初始化变量
    sess.run(tf.global_variables_initializer())

    # 将数据入队
    sess.run(enqueue_op)

    # 保存队列的状态
    saver.save(sess, "/path/to/save/queue_state.ckpt")

# 恢复队列的状态
with tf.Session() as sess:
    # 初始化变量
    sess.run(tf.global_variables_initializer())

    # 恢复队列的状态
    saver.restore(sess, "/path/to/save/queue_state.ckpt")

    # 从队列中获取数据
    dequeue_op = queue.dequeue()
    data = sess.run(dequeue_op)
    print(data)