pytorch學習筆記（六）——pytorch進階教程之tensor高階操作

pytorch學習筆記（六）——pytorch進階教程之高階操作 目錄 where gather 目錄 where 新生成的tensor取決與輸入x，y和條件condition。condition是一個矩陣，如果元素是1對應x，元素是0對應y。 示例代碼： cond>0.5返回的是一個[2,2]size的矩陣，大於0.5對應元素爲1，否則爲0。 只有最右下角元素不成立返回0，所以where函數得到

>>阅读原文<<