网络编程
位置：首页>> 网络编程>> Python编程>> tensorflow 获取所有variable或tensor的name示例

tensorflow 获取所有variable或tensor的name示例

作者：_沥川往事　　发布时间：2021-04-02 22:29:54　

标签：tensorflow,variable,tensor,name

获取所有variable（每个op中可训练的张量）的name：


for variable_name in tf.global_variables():
 print(variable_name)

获取所有tensor（每个op的输出张量）的name：


for tensor_name in tf.contrib.graph_editor.get_tensors(tf.get_default_graph()):
 print(tensor_name)

获取所有op及其输入输出的name：


with tf.Session() as sess:
 for node in sess.graph_def.node:
   print(node)

来源：https://blog.csdn.net/yuejisuo1948/article/details/88858190

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com