关于pytorch:pytorch前向传播和反向传播

43次阅读

共计 297 个字符,预计需要花费 1 分钟才能阅读完成。

如下程序

import torch

a = torch.tensor(1.0, requires_grad=True)
b = torch.tensor(1.0, requires_grad=True)
print(a.grad)
y = (a ** 2) * (b ** 3)
z = y ** 100
print(z)
z.backward()
print(a.grad.data)
print(b.grad.data)

将会打印

None
tensor(1., grad_fn=<PowBackward0>)
tensor(200.)
tensor(300.)
前向流传就是 z 的值是从 a b 计算 y 的值 再从 y 计算 z 的值
反向流传就是从 z 对 a b 的表达式中计算对应的偏导

正文完
 0