2020-09-27
对Pytorch中backward()函数参数的理解y.backward(v)代码验证运行结果y.backward(v)以y = w x为例,计算x关于y的梯度: y对x的导数,包含了y中各个元素对x的导数,所以,计算时y中各个元素对x的偏导需要一个权重,因此,v需要和y同维度。代码验证import torchx = torch.tensor([[[1.],[3.]]],requires_grad = True) #requires_grad打开梯度计算,必须是浮点数W = torch.ten
原创
2020-09-27 10:42:03 ·
136 阅读 ·
1 评论