溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章主要講解了Pytorch高階OP怎么操作where，gather，內容清晰明了，對此有興趣的小伙伴可以學習一下，相信大家閱讀完之后會有幫助。
PyTorch是一個非常有可能改變深度學習領域前景的Python庫。我嘗試使用了幾星期PyTorch，然后被它的易用性所震驚，在我使用過的各種深度學習庫中，PyTorch是最靈活、最容易掌握的。
一、where
1）torch.where(condition, x, y) # condition是條件，滿足條件就返回x，不滿足就返回y
2）特點，相比for循環(huán)的優(yōu)點是：可以布置在GPU上運行
二、gather
1）官方解釋：根據指定的維度和索引值來篩選值
2）舉例
看完上述內容，是不是對Pytorch高階OP怎么操作where，gather有進一步的了解，如果還想學習更多內容，歡迎關注億速云行業(yè)資訊頻道。
免責聲明：本站發(fā)布的內容（圖片、視頻和文字）以原創(chuàng)、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
