如何通过名字获得tensorflow操作?

2024-06-26 00:11:03 发布

您现在位置:Python中文网/ 问答频道 /正文

可以用tf.get_default_graph().get_tensor_by_name("tensor_name:0")按名称获取张量

但是,您能得到一个操作,比如队列上的Optimizer.minimize,或者enqueue操作吗?

在我的第一个模型中,我从build_model函数返回了所有需要的张量和ops。但是张量的列表很难看。在后来的模型中,我把所有的张量和运算放入字典中,以便更容易地访问。这一次,我想只要我需要的话,我就直接查张量的名字,但我不知道如何在行动中做到这一点。

还是有更好的办法?我发现所有地方都需要各种张力和操作。培训、推理代码、测试用例,因此需要一种很好的标准方法来访问图形的各个部分,而不必到处传递变量。


Tags: name模型build名称defaultgetbymodel