tensorflow获取模型中tensor和node的语句

加载保存好的模型之后,有时候需要知道模型中所有的tensor或者所有的node的名称,以备做进一步的查询。很多资料说的其实不太准确,这两种经常混到一起。参考资料——stackoverflow上的一个同类问题。

1.获取模型中所有的node

[n.name for n in tf.get_default_graph().as_graph_def().node]

2.获取模型中所有的tensor

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]

3.根据名称获得某个tensor或者op

##根据名称获得tensor
tf.get_default_graph().get_tensor_by_name('tensor_name')
##根据名称获得op
tf.get_default_graph().get_operation_by_name('op_name')

你可能感兴趣的:(Python,TensorFlow)