TensorFlow查看输入节点和输出节点名称方式
TensorFlow定义输入节点名称input_name：
withtf.name_scope('input'):
bottleneck_input=tf.placeholder_with_default(
bottleneck_tensor,
shape=[batch_size,bottleneck_tensor_size],
name='Mul')
TensorFlow查看pb数据库里面的输入节点和输出节点：
importtensorflowastf importos model_dir='./tmp/' model_name='output_graph.pb' defcreate_graph(): withtf.gfile.FastGFile(os.path.join( model_dir,model_name),'rb')asf: graph_def=tf.GraphDef() graph_def.ParseFromString(f.read()) tf.import_graph_def(graph_def,name='') create_graph() tensor_name_list=[tensor.namefortensorintf.get_default_graph().as_graph_def().node] fortensor_nameintensor_name_list: print(tensor_name,'\n')
以上这篇TensorFlow查看输入节点和输出节点名称方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。