pytorch中backward()函数详解

在最开始使用Pytorch时,关于反向传播函数,有时会报错,简单的说,如果你的Y值是个标量,那么直接使用:

y.backward()

就可以了,但是如果你的Y是个向量或矩阵,那么就不一样:

y.backward(torch.ones_like(x))

具体原因如下:

(19条消息) pytorch中backward()函数详解_Camlin_Z的博客-CSDN博客

pytorch中backward()函数详解_第1张图片 

 

你可能感兴趣的:(pytorch,深度学习,神经网络)