如何从TensorFlow图中获取Queue对象?

时间:2017-12-05 00:17:01

标签: python tensorflow

在TensorFlow中,我有一个图表,里面有一个string_input_producer,用在它的输入管道中。

我正在从检查点文件加载图形,因此在创建原始对象时无法访问该对象。但是,我需要运行此对象的入队方法。

我尝试使用get_operation_by_name和get_tensor_by_name来获取FIFOQueue对象,这显然不起作用,因为队列既不是操作也不是张量。是否有像上面提到的功能可以做我想要的? (假的,例如get_queue_by_name)我怎么能解决我的问题?

1 个答案:

答案 0 :(得分:0)

我认为到目前为止,这是不可能的。但是,您可以使用get_operation_by_name来获取队列对象的方法。例如get_operation_by_name("queue_name/enqueue")。您还可以创建自己的Queue对象,该对象在内部简单地调用相应的图形操作。