說明
本內容節選自《深度學習框架pytorch》，目前本專欄限時打折中，歡迎訂閱
正文
本文將介紹在使用深度學習框架pytorch中的兩個常用的函數，這兩個函數在搭建模型的時候會經常用到，下面我們看一下它的具體用法
torch.where(condition,x,y)
首先condition和x、y他們三個的shape應該是一致的
然後這個方法的意思就是說判斷condition，如果conditon為1，則取x，否則取y
gather就是一個查表得過程
torch.gather(input,dim,index,out=None)
這個表示在input的dim維度查找索引index所對應的值
分享到:
閱讀更多 幻風的AI之路 的文章
