一、where
1）torch.where(condition, x, y) # condition是條件，滿足條件就返回x，不滿足就返回y
2）特點，相比for循環的優點是：可以布置在GPU上運行
二、gather
1）官方解釋：根據指定的維度和索引值來篩選值
2）舉例
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
PyTorch——(4)where條件判斷、gather查表
Pytorch張量高階操作
[深度學習] pytorch學習筆記（1）(數據類型、基礎使用、自動求導、矩陣操作、維度變換、廣播、拼接拆分、基本運算、范數、argmax、矩陣比較、where、gather)
Pytorch的gather用法理解
PyTorch中scatter和gather的用法
pytorch 中的 gather() 函數詳解
關於Pytorch的二維tensor的gather和scatter_操作用法分析
Pytorch中的torch.gather函數
tensorflow添加新操作（Op）
tf.gather()、tf.gather_nd()、tf.batch_gather()、tf.where()和tf.slice()