可以用tf.get_default_graph().get_tensor_by_name("tensor_name:0")
按名称获取张量
但是,您能得到一个操作,比如队列上的Optimizer.minimize
,或者enqueue
操作吗?
在我的第一个模型中,我从build_model
函数返回了所有需要的张量和ops。但是张量的列表很难看。在后来的模型中,我把所有的张量和运算放入字典中,以便更容易地访问。这一次,我想只要我需要的话,我就直接查张量的名字,但我不知道如何在行动中做到这一点。
还是有更好的办法?我发现所有地方都需要各种张力和操作。培训、推理代码、测试用例,因此需要一种很好的标准方法来访问图形的各个部分,而不必到处传递变量。
可以使用^{} 方法按名称获取
tf.Operation
。例如,要从默认图中获取名为"enqueue"
的操作:相关问题 更多 >
编程相关推荐