您可以用tf.get_default_graph().get_tensor_by_name("tensor_name:0")
但是您可以在队列上获得操作,例如Optimizer.minimize
或enqueue
操作?
在我的第一个型号中,我返回了build_model
功能所需的所有张量和操作。但是张量的清单变得丑陋。在后来的型号中,我将所有张量和操作扔到字典中,以便于访问。这次,我以为我只是按名称查找张量,但我不知道该如何使用操作。
还是有更好的方法?我发现到处都需要各种张量和操作。培训,推理代码,测试用例,因此渴望访问图的各个部分而不经过变量的愿望。
您可以使用tf.Graph.get_operation_by_name()
方法来获取tf.Operation
。例如,从默认图中获取称为"enqueue"
的操作:
op = tf.get_default_graph().get_operation_by_name("enqueue")