当前位置：首页 > article >正文

详解Pytorch：张量自动微分

article 2025/3/3 7:01:48

张量自动微分（Automatic Differentiation, AD） 是 PyTorch 的核心功能之一，它允许我们自动计算张量的梯度。自动微分在深度学习中非常重要，因为神经网络的训练依赖于梯度下降法，而梯度下降法需要计算损失函数对模型参数的梯度。

自动微分的基本概念

什么是自动微分？
- 自动微分是一种计算函数导数（梯度）的技术。
- 它通过构建计算图（Computation Graph）来跟踪张量的操作，并在反向传播时自动计算梯度。
为什么需要自动微分？
- 在深度学习中，模型的参数通常通过梯度下降法进行优化。
- 手动计算梯度非常繁琐且容易出错，尤其是对于复杂的模型。
- 自动微分可以高效且准确地计算梯度。
计算图（Computation Graph）
- 计算图是 PyTorch 自动微分的基础。
- 每个张量操作（如加法、乘法）都会在计算图中创建一个节点。
- 计算图记录了张量之间的依赖关系，使得反向传播时可以自动计算梯度。

自动微分的实现

在 PyTorch 中，自动微分通过以下步骤实现：

设置 requires_grad=True：
- 将张量的 requires_grad 属性设置为 True，表示需要计算该张量的梯度。
```
x = torch.tensor([1.0, 2.0, 3.0], requires_grad=True)
```
定义计算图：
- 对张量进行操作时，PyTorch 会自动构建计算图。
```
y = x * 2
z = y.sum()
```
反向传播：
- 调用 .backward() 方法，自动计算梯度。
```
z.backward()
```
获取梯度：
- 通过 .grad 属性获取张量的梯度。
```
print(x.grad)  # 输出: tensor([2., 2., 2.])
```

示例：自动微分的基本用法

以下是一个简单的示例，展示如何使用 PyTorch 进行自动微分：

import torch

# 创建一个需要计算梯度的张量
x = torch.tensor([1.0, 2.0, 3.0], requires_grad=True)

# 定义计算图
y = x * 2  # y = [2, 4, 6]
z = y.sum()  # z = 2 + 4 + 6 = 12

# 反向传播，计算梯度
z.backward()

# 获取梯度
print(x.grad)  # 输出: tensor([2., 2., 2.])

自动微分的详细步骤

1. 前向传播

在前向传播过程中，PyTorch 会记录所有张量操作，并构建计算图。

例如：

x = torch.tensor([1.0, 2.0, 3.0], requires_grad=True)
y = x * 2
z = y.sum()

2. 反向传播

在反向传播过程中，PyTorch 会从输出张量开始，沿着计算图反向传播梯度。
例如：
```
z.backward()
```

3. 梯度计算

对于每个需要计算梯度的张量，PyTorch 会计算其梯度并存储在 .grad 属性中。

例如：

print(x.grad)  # 输出: tensor([2., 2., 2.])

自动微分的应用场景

神经网络的训练：

在神经网络中，自动微分用于计算损失函数对模型参数的梯度。

例如：

# 定义模型
model = torch.nn.Linear(10, 1)
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)

# 前向传播
output = model(input)
loss = torch.nn.functional.mse_loss(output, target)

# 反向传播
loss.backward()

# 更新参数
optimizer.step()

自定义函数的梯度计算：

自动微分可以用于计算任意函数的梯度。

例如：

def my_function(x):
    return x ** 2 + 3 * x + 1

x = torch.tensor(2.0, requires_grad=True)
y = my_function(x)
y.backward()
print(x.grad)  # 输出: tensor(7.)

注意事项

梯度累积：
- 每次调用 .backward() 时，梯度会累积到 .grad 属性中。
- 如果需要清除梯度，可以调用 .zero_grad() 方法。
```
x.grad.zero_()
```

非标量输出：

如果输出是一个非标量张量（如向量或矩阵），需要为 .backward() 提供一个 gradient 参数。

x = torch.tensor([1.0, 2.0, 3.0], requires_grad=True)
y = x * 2
y.backward(torch.tensor([1.0, 1.0, 1.0]))  # 提供梯度
print(x.grad)  # 输出: tensor([2., 2., 2.])