加载保存好的模型之后,有时候需要知道模型中所有的tensor或者所有的node的名称,以备做进一步的查询。很多资料说的其实不太准确,这两种经常混到一起。参考资料——stackoverflow上的一个同类问题。
[n.name for n in tf.get_default_graph().as_graph_def().node]
graph = tf.get_default_graph()
tensors_per_node = [node.values() for node in graph.get_operations()]
tensor_names = [tensor.name for tensors in tensors_per_node for tensor in tensors]
##根据名称获得tensor
tf.get_default_graph().get_tensor_by_name('tensor_name')
##根据名称获得op
tf.get_default_graph().get_operation_by_name('op_name')