qq_40507137的博客

原创 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 torch x = torch.tensor([[[1.],[3.]]],requires_grad = True) #requires_grad打开梯度计算，必须是浮点数 W = torch.ten