pytorch 常用函數(shù) max ,eq說(shuō)明
更新時(shí)間：2020年06月28日 09:37:32 作者：藍(lán)鯨123
這篇文章主要介紹了pytorch 常用函數(shù) max eq說(shuō)明，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
max找出tensor 的行或者列最大的值：
找出每行的最大值:
import torch outputs=torch.FloatTensor([[1],[2],[3]]) print(torch.max(outputs.data,1))
輸出：
(tensor([ 1., 2., 3.]), tensor([ 0, 0, 0]))
找出每列的最大值：
import torch outputs=torch.FloatTensor([[1],[2],[3]]) print(torch.max(outputs.data,0))
輸出結(jié)果：
(tensor([ 3.]), tensor([ 2]))
Tensor比較eq相等：
import torch outputs=torch.FloatTensor([[1],[2],[3]]) targets=torch.FloatTensor([[0],[2],[3]]) print(targets.eq(outputs.data))
輸出結(jié)果：
tensor([[ 0], [ 1], [ 1]], dtype=torch.uint8)
使用sum() 統(tǒng)計(jì)相等的個(gè)數(shù)：
import torch outputs=torch.FloatTensor([[1],[2],[3]]) targets=torch.FloatTensor([[0],[2],[3]]) print(targets.eq(outputs.data).cpu().sum())
輸出結(jié)果：
tensor(2)
補(bǔ)充知識(shí)：PyTorch - torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le
flyfish
torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le
以上全是簡(jiǎn)寫(xiě)
參數(shù)是input, other, out=None
逐元素比較input和other
返回是torch.BoolTensor
import torch a=torch.tensor([[1, 2], [3, 4]]) b=torch.tensor([[1, 2], [4, 3]]) print(torch.eq(a,b))#equals # tensor([[ True, True], # [False, False]]) print(torch.ne(a,b))#not equal to # tensor([[False, False], # [ True, True]]) print(torch.gt(a,b))#greater than # tensor([[False, False], # [False, True]]) print(torch.lt(a,b))#less than # tensor([[False, False], # [ True, False]]) print(torch.ge(a,b))#greater than or equal to # tensor([[ True, True], # [False, True]]) print(torch.le(a,b))#less than or equal to # tensor([[ True, True], # [ True, False]])
以上這篇pytorch 常用函數(shù) max ,eq說(shuō)明就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python快速實(shí)現(xiàn)一個(gè)線程池的示例代碼
當(dāng)有多個(gè)?IO?密集型的任務(wù)要被處理時(shí)，我們自然而然會(huì)想到多線程。而線程池的實(shí)現(xiàn)也很簡(jiǎn)單，因?yàn)?Python?提供了一個(gè)標(biāo)準(zhǔn)庫(kù)?concurrent.futures，已經(jīng)內(nèi)置了對(duì)線程池的支持。所以本篇文章，我們就來(lái)詳細(xì)介紹一下該模塊的用法2022-07-07
Django學(xué)習(xí)教程之靜態(tài)文件的調(diào)用詳解
這篇文章主要給大家介紹了關(guān)于Django學(xué)習(xí)教程之靜態(tài)文件調(diào)用的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家學(xué)習(xí)或者使用django具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2018-05-05
這篇文章主要介紹了Python常用的文件及文件路徑、目錄操作方法匯總介紹,本文集合了最常用的一些文件和目錄操作函數(shù),并一一介紹它們的作用,需要的朋友可以參考下2015-05-05
這篇文章主要介紹了python刪除列表中重復(fù)記錄的方法,涉及Python操作列表的相關(guān)技巧,需要的朋友可以參考下2015-04-04
這篇文章主要介紹了簡(jiǎn)單了解python數(shù)組的基本操作,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-11-11
Python利用scikit-learn實(shí)現(xiàn)近鄰算法分類(lèi)的示例詳解
scikit-learn已經(jīng)封裝好很多數(shù)據(jù)挖掘的算法，這篇文章就來(lái)用scikit-learn實(shí)現(xiàn)近鄰算法分類(lèi)，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以了解一下2023-02-02
Python+matplotlib實(shí)現(xiàn)折線圖的美化
這篇文章主要和大家分享一個(gè)非常有趣的Python教程—如何美化一個(gè)?matplotlib折線圖。文中的示例代碼講解詳細(xì)，感興趣的可以了解一下2022-05-05
這篇文章主要介紹了教你用Python下載抖音無(wú)水印視頻,文中有非常詳細(xì)的代碼示例,對(duì)正在學(xué)習(xí)python的小伙伴們有很好地幫助,需要的朋友可以參考下2021-05-05
UI自動(dòng)化定位常用實(shí)現(xiàn)方法代碼示例
這篇文章主要介紹了UI自動(dòng)化定位常用實(shí)現(xiàn)方法代碼示例,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-10-10
python利用MethodType綁定方法到類(lèi)示例代碼
這篇文章主要給大家介紹了關(guān)于python利用MethodType綁定方法到類(lèi)的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面跟著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧。2017-08-08