脚本专栏 
首页 > 脚本专栏 > 浏览文章

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

(编辑：jimmy 日期: 2025/1/18 浏览：3 次 )

获取所有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)

以上这篇tensorflow 获取所有variable或tensor的name示例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。

上一篇：TensorFlow tf.nn.max_pool实现池化操作方式
下一篇：TensorFlow tf.nn.conv2d实现卷积的方式
一句话新闻
微软与英特尔等合作伙伴联合定义“AI PC”：键盘需配有Copilot物理按键
几个月来，英特尔、微软、AMD和其它厂商都在共同推动“AI PC”的想法，朝着更多的AI功能迈进。在近日，英特尔在台北举行的开发者活动中，也宣布了关于AI PC加速计划、新的PC开发者计划和独立硬件供应商计划。
在此次发布会上，英特尔还发布了全新的全新的酷睿Ultra Meteor Lake NUC开发套件，以及联合微软等合作伙伴联合定义“AI PC”的定义标准。