理解: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
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容