21、TensorFlow 基础与线性回归实战

TensorFlow 基础与线性回归实战

1. TensorFlow 基本操作

1.1 会话管理

在 TensorFlow 中，会话（Session）用于执行计算图。可以使用 InteractiveSession 来创建一个交互式会话，但使用完后需要手动关闭：

import tensorflow as tf

sess = tf.InteractiveSession()
init = tf.global_variables_initializer()
init.run()
f = tf.constant(42)
result = f.eval()
print(result)
sess.close()

1.2 计算图管理

TensorFlow 程序通常分为两个阶段：构建计算图（构建阶段）和运行计算图（执行阶段）。构建阶段构建表示机器学习模型和训练所需计算的计算图，执行阶段则反复评估训练步骤以逐步改进模型参数。

默认情况下，创建的任何节点都会自动添加到默认图中：

x1 = tf.Variable(1)
print(x1.graph is tf.get_default_graph())  # 输出: True

如果需要管理多个独立的图，可以创建一个新的图并在 with 块中临时将其设置为默认图：