PyTorch中怎么进行梯度计算和反向传播

pytorch

2382

2024/2/19 13:31:24

栏目: 深度学习

在PyTorch中，可以通过torch.autograd模块来进行梯度计算和反向传播。以下是一个简单的示例代码：

import torch

# 创建一个张量并设置requires_grad=True来追踪其计算历史
x = torch.tensor([2.0], requires_grad=True)
y = torch.tensor([3.0], requires_grad=True)

# 定义一个计算图
z = x * y

# 计算z相对于x和y的梯度
z.backward()

# 访问x和y的梯度
print(x.grad)  # 输出tensor([3.])
print(y.grad)  # 输出tensor([2.])

在这个示例中，我们首先创建了两个张量x和y，并设置requires_grad=True来追踪它们的计算历史。然后我们定义了一个计算图z = x * y，然后调用z.backward()来计算z相对于x和y的梯度。最后我们可以通过访问x.grad和y.grad来获取梯度值。

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

PyTorch中怎么进行梯度计算和反向传播

最新知识库

相关标签