2016-06-21 31 views
6

按照標題。我想利用這種操作來重命名節點並更好地組織圖。還是有其他推薦的做法來重命名圖中的現有節點嗎?謝謝!張量流中是否存在無操作(傳遞)操作?

+2

爲了完整起見,有一個'tf.no_op' https://www.tensorflow.org/versions/master/api_docs/python/control_flow_ops.html#no_op,但我想你想要一個像' tf.identity' – Falcon

+0

經過一番探索之後,我發現使用'collections'比重命名更有幫助。看看這裏的例子:https://www.tensorflow.org/versions/r0.9/how_tos/meta_graph/index.html – Falcon

回答

4

據我所知,你不能重命名曾經創建的張量。

但是,您可以使用額外的 「不操作」 的操作(就像你說的):

  • tf.Tensortf.identity(input_tensor, name='your_new_name')

  • 一個操作:tf.group(input_operation, name='your_new_name')


之後,您可以撥打input_tensor有:

graph = tf.get_default_graph() 
graph.get_tensor_by_name('your_new_name:0') 

還是input_operation有:

graph = tf.get_default_graph() 
graph.get_operation_by_name('your_new_name') 
9

tf.no_op它允許你添加哪些什麼都不做的操作。

相關問題