理解:Tensor.backward()中当tensor不是标量的时候,需要一个与tensor同等形状的入参tensor。

在PyTorch的backward模块里,如果tensor是一个vector不是scalar,需要入参里添加要给和tensor(这里用y表示,y_0, y_1, y_2)相同形状的vector(这里使用w表示, w_0, w_1, w_2)。针对这一现象,做出一些解释。

基础代码如下:


代码.png

解释如下:


笔记.jpg

推荐阅读更多精彩内容