Pytorch高階OP操作where,gather原理
更新時間:2020年04月30日 11:16:00 作者:杰鑫哥
這篇文章主要介紹了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)舉例
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關文章
Tensorflow中k.gradients()和tf.stop_gradient()用法說明
這篇文章主要介紹了Tensorflow中k.gradients()和tf.stop_gradient()用法說明,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06pandas?dataframe獲取所有行名稱與列名稱方法示例
這篇文章主要給大家介紹了關于pandas?dataframe獲取所有行名稱與列名稱的相關資料,Pandas是Python中用于數據分析的非常重要的庫,它提供了多種方法來獲取列名,需要的朋友可以參考下2023-09-09在pandas中遍歷DataFrame行的實現(xiàn)方法
這篇文章主要介紹了在pandas中遍歷DataFrame行的實現(xiàn)方法,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2019-10-10