如何按名称获得TensorFlow OP



您可以用tf.get_default_graph().get_tensor_by_name("tensor_name:0")

获得张量

但是您可以在队列上获得操作,例如Optimizer.minimizeenqueue操作?

在我的第一个型号中,我返回了build_model功能所需的所有张量和操作。但是张量的清单变得丑陋。在后来的型号中,我将所有张量和操作扔到字典中,以便于访问。这次,我以为我只是按名称查找张量,但我不知道该如何使用操作。

还是有更好的方法?我发现到处都需要各种张量和操作。培训,推理代码,测试用例,因此渴望访问图的各个部分而不经过变量的愿望。

您可以使用tf.Graph.get_operation_by_name()方法来获取tf.Operation。例如,从默认图中获取称为"enqueue"的操作:

op = tf.get_default_graph().get_operation_by_name("enqueue")

最新更新