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