pytorch實現(xiàn)onehot編碼轉(zhuǎn)為普通label標簽
更新時間：2020年01月02日 09:16:10 作者：swuxyj
今天小編就為大家分享一篇pytorch實現(xiàn)onehot編碼轉(zhuǎn)為普通label標簽，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
label轉(zhuǎn)onehot的很多，但是onehot轉(zhuǎn)label的有點難找，所以就只能自己實現(xiàn)以下，用的topk函數(shù)，不知道有沒有更好的實現(xiàn)
one_hot = torch.tensor([[0,0,1],[0,1,0],[0,1,0]]) print(one_hot) label = torch.topk(one_hot, 1)[1].squeeze(1) print(label)
tensor([[0, 0, 1], [0, 1, 0], [0, 1, 0]]) tensor([2, 1, 1])
以上這篇pytorch實現(xiàn)onehot編碼轉(zhuǎn)為普通label標簽就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
下面小編就為大家分享一篇pandas 獲取季度,月度,年度首尾日期的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
今天小編就為大家分享一篇python: 自動安裝缺失庫文件的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-10-10
Python如何統(tǒng)計大小寫字母個數(shù)和數(shù)字個數(shù)
這篇文章主要介紹了Python如何統(tǒng)計大小寫字母個數(shù)和數(shù)字個數(shù)問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
使用python操作lmdb對數(shù)據(jù)讀取的實例
這篇文章主要介紹了使用python操作lmdb對數(shù)據(jù)讀取的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12
這篇文章主要介紹了Python3壓縮和解壓縮實現(xiàn)代碼,代碼簡單易懂，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-03-03
這篇文章主要介紹了python 繪制國旗的示例，幫助大家利用python繪制，處理圖像，感興趣的朋友可以了解下2020-09-09
這篇文章主要為大家詳細介紹了使用python3.5仿微軟記事本notepad的相關(guān)資料，具有一定的參考價值，感興趣的小伙伴們可以參考一下2016-06-06
python函數(shù)存儲在模塊的優(yōu)點及用法總結(jié)
在本篇文章里小編給大家整理了一篇關(guān)于python函數(shù)存儲在模塊的優(yōu)點及用法相關(guān)內(nèi)容，有興趣的朋友們可以跟著學(xué)習(xí)下。2021-10-10
Python3+selenium實現(xiàn)cookie免密登錄的示例代碼
這篇文章主要介紹了Python3+selenium實現(xiàn)cookie免密登錄的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-03-03
python開發(fā)之IDEL(Python GUI)的使用方法圖文詳解
這篇文章主要介紹了python開發(fā)之IDEL(Python GUI)的使用方法,結(jié)合圖文形式較為詳細的分析總結(jié)了Python GUI的具體使用方法,需要的朋友可以參考下2015-11-11