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

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

基础代码如下:


代码.png

解释如下:


笔记.jpg

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