理解:Tensor.backward()中當tensor不是標量的時候,需要一個與tensor同等形狀的入?yún)ensor。

在PyTorch的backward模塊里,如果tensor是一個vector不是scalar,需要入?yún)⒗锾砑右o和tensor(這里用y表示,y_0, y_1, y_2)相同形狀的vector(這里使用w表示, w_0, w_1, w_2)。針對這一現(xiàn)象,做出一些解釋。

基礎代碼如下:


代碼.png

解釋如下:


筆記.jpg
最后編輯于
?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容