8 操作与计算图

查看大图

TensorFlow 学习要把张量、模型、训练和部署放在一条线上，而不是只看单段代码。阅读时可以按「操作的概念 -> 操作类型 -> 示例：基本数学操作 -> 计算图的概念」建立结构，再回到正文里的代码、案例或指标做验证。

查看大图

读完后，用一个真实小任务复查：输入是什么，处理环节在哪里，输出是否可验收；失败时先查「操作的概念」，再查「操作类型」。

在本章节中，我们将深入探讨 TensorFlow 中的操作（Operations）和计算图（Computational Graphs）。在学习了 Tensors 的基本概念后，理解操作与计算图对于充分利用 TensorFlow 进行机器学习和深度学习至关重要。

1. 操作的概念

在 TensorFlow 中，操作是指在 Tensors 之间执行的计算，可以将其视为图中的节点。每个操作都接收输入 Tensors，并输出一个或多个 Tensors。操作可以是数学计算、数据变换、模型训练等。

查看大图

理解 TensorFlow 操作与计算图时，先看张量如何流经操作节点，再看依赖关系如何决定执行顺序。

1.1 操作类型

TensorFlow 中的操作类型繁多，以下是一些常见的操作类型：

• 数学操作: 如加法、减法、乘法和除法。
• 逻辑操作: 如与、或和非等。
• 数组操作: 如重塑、拼接和切片等。
• 优化操作: 如梯度下降等学习算法。

1.2 示例：基本数学操作

下面我们通过一些简单的代码示例，展示如何在 TensorFlow 中使用基础操作。首先，我们需要导入 TensorFlow：

import tensorflow as tf

接下来，我们将创建一些 Tensors 并进行基本的数学运算：

# 创建 Tensor
a = tf.constant(5)
b = tf.constant(3)

# 加法操作
add_result = tf.add(a, b)

# 乘法操作
mul_result = tf.multiply(a, b)

print("加法结果:", add_result.numpy())
print("乘法结果:", mul_result.numpy())

在上面的代码中，我们创建了两个常量 Tensor a 和 b，并使用 tf.add 和 tf.multiply 进行了加法和乘法操作。

2. 计算图的概念

计算图是 TensorFlow 的核心构建块。它是操作和 Tensors 的图形表示。在计算图中，节点代表操作，而边代表 Tensors。这个设计使得 TensorFlow 可以有效地组织和优化计算。

查看大图

《操作与计算图》可以按“场景、概念、动作、结果”来读。先把这四件事对齐，再回到正文里的参数、代码或流程。

2.1 构建计算图

构建计算图的过程通常包括以下几个步骤：

1. 创建 Tensors。
2. 定义操作。
3. 将操作连接到 Tensors。

通过计算图，TensorFlow 可以将计算转化为图的遍历，从而高效地计算结果。在 TensorFlow 中，构建计算图不会立即执行，而是定义了一个操作的“计划”。我们可以在之后的会话中运行这个图。

2.2 示例：构建简单计算图

以下是一个创建简单计算图的例子：

# 创建一张计算图
@tf.function
def simple_graph(x, y):
    return tf.add(x, y), tf.multiply(x, y)

# 定义输入
x = tf.constant(10)
y = tf.constant(20)

# 执行图
add_result, mul_result = simple_graph(x, y)

print("加法结果:", add_result.numpy())
print("乘法结果:", mul_result.numpy())

在这个示例中，我们创建了一个名为 simple_graph 的函数，使用 @tf.function 装饰器来指示 TensorFlow 创建一个计算图。当我们传入 Tensors x 和 y 时，这个函数会定义加法和乘法两个操作，而后执行并返回结果。

查看大图

读到这里，可以把《操作与计算图》整理成一张复盘表：先说清主线，再拿一个小任务检查结果。

查看大图

读完《操作与计算图》后，可以先挑一个小样例走完整流程，再判断哪些步骤已经能独立完成。

3. 总结

在这一部分中，我们了解了 TensorFlow 中的操作和计算图。操作是执行具体计算的节点，而计算图则是组织这些操作和 Tensors 的结构。通过理解和运用这些概念，我们能够高效地构建和运行 TensorFlow 程序，为后续的模型训练和数据处理打下基础。

在接下来的章节中，我们将学习会话（Sessions）的使用，这是执行计算图的具体方式，帮助我们真正运行和测试我们的 TensorFlow 程序。