pytorch前向传播和反向传播
Posted o_ra
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch前向传播和反向传播相关的知识，希望对你有一定的参考价值。
如下程序
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的表达式中计算对应的偏导
以上是关于pytorch前向传播和反向传播的主要内容，如果未能解决你的问题，请参考以下文章