ShusenTang / Dive-into-DL-PyTorch

本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。

http://tangshusen.me/Dive-into-DL-PyTorch

Apache License 2.0

18.07k stars 5.37k forks source link

Open zthcool opened 2 years ago

zthcool commented 2 years ago

bug描述 在按照教程2.3.3梯度章节中 “

再来反向传播一次，注意grad是累加的

out2 = x.sum() out2.backward() print(x.grad) ” 时我的返回结果是 tensor([[1., 1.], [1., 1.]]) 如图请问是我的输入有问题么？

版本信息 pytorch: 1.9.0 torchvision: 0.10.0 torchtext: None ...

dingtao-shen commented 2 years ago

你backward的是out2，对应的是x.sum()，梯度是1没问题；原章节里的操作是对out = z.mean()进行backward之后查看x.grad