【PyTorch】RuntimeError: one of the variables needed for gradient computation has been modified by an

最新推荐文章于 2024-07-02 09:32:38 发布

ncc1995

最新推荐文章于 2024-07-02 09:32:38 发布

阅读量7.6w

点赞数 67

分类专栏： pytorch 深度学习

本文链接：https://blog.csdn.net/ncc1995/article/details/99542594

版权

错误：

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation

报错原因：

import torch
x = torch.FloatTensor([[1., 2.]])
w1 = torch.FloatTensor([[2.], [1.]])
w2 = torch.FloatTensor([3.])
w1.requires_grad = True
w2.requires_grad = True

d = torch.matmul(x, w1)
f = torch.matmul(d, w2)
d[:] = 1 # 因为这句, 代码报错了 RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation

f.backward()

因为 $f = matmul(d, w2), \frac{\partial f}{\partial w2} = g(d),$ 对于