在PyTorch的backward模塊里,如果tensor是一個vector不是scalar,需要入?yún)⒗锾砑右o和tensor(這里用y表示,,
,
)相同形狀的vector(這里使用w表示,
,
,
)。針對這一現(xiàn)象,做出一些解釋。
基礎代碼如下:
代碼.png
解釋如下:
筆記.jpg
在PyTorch的backward模塊里,如果tensor是一個vector不是scalar,需要入?yún)⒗锾砑右o和tensor(這里用y表示,,
,
)相同形狀的vector(這里使用w表示,
,
,
)。針對這一現(xiàn)象,做出一些解釋。
基礎代碼如下:
代碼.png
解釋如下:
筆記.jpg