TI 怎么把2个变量都在一个graph中显示
时间: 2023-11-28 08:47:39 浏览: 48
您可以使用 TensorFlow 的 `tf.summary` 模块来记录和可视化多个变量在同一个图中的值。以下是一个示例:
```python
import tensorflow as tf
# 创建两个变量
var1 = tf.Variable(1.0, name="var1")
var2 = tf.Variable(2.0, name="var2")
# 创建一个操作,将变量相加
sum_vars = tf.add(var1, var2, name="sum_vars")
# 创建一个 FileWriter 对象来将日志写入磁盘
writer = tf.summary.FileWriter('./logs')
# 创建一个合并所有变量摘要的操作
merged_summary = tf.summary.merge_all()
# 在会话中运行变量初始化操作
init = tf.global_variables_initializer()
with tf.Session() as sess:
sess.run(init)
# 运行 merged_summary 操作并将结果写入日志
summary = sess.run(merged_summary)
writer.add_summary(summary)
# 运行 sum_vars 操作并将结果打印出来
result = sess.run(sum_vars)
print(result)
```
在这个示例中,我们创建了两个变量 `var1` 和 `var2`,然后创建了一个操作 `sum_vars`,将这两个变量相加。接下来,我们创建了一个 FileWriter 对象来将日志写入磁盘,然后创建了一个合并所有变量摘要的操作 `merged_summary`。在会话中运行变量初始化操作,然后运行 merged_summary 操作,并将结果写入日志。最后,我们运行 sum_vars 操作并将结果打印出来。
注意,这个示例中只有一个操作(sum_vars),但是您可以将多个操作添加到摘要中,并在会话中运行 merged_summary 操作来记录和可视化所有变量的值。
阅读全文