pytorch標簽轉(zhuǎn)onehot形式實例
更新時間：2020年01月02日 09:12:06 作者：頭發(fā)光了你就強了
今天小編就為大家分享一篇pytorch標簽轉(zhuǎn)onehot形式實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
代碼：
import torch class_num = 10 batch_size = 4 label = torch.LongTensor(batch_size, 1).random_() % class_num print(label.size()) one_hot = torch.zeros(batch_size, class_num).scatter_(1, label, 1) print(one_hot)
輸出：
torch.Size([4, 1]) tensor([[0., 0., 0., 0., 0., 0., 0., 1., 0., 0.], [0., 0., 0., 0., 0., 1., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 1., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0., 0., 1.]])
注意：
label的形狀必須是[n,1]的，也就是必須是二維的，且第二個維度長度為1，如果是一維度的，則需要升維度，代碼如下：
import torch class_num = 10 batch_size = 4 label = torch.LongTensor(batch_size).random_() % class_num print(label.size()) label = torch.unsqueeze(label,dim=1) print(label.size())
以上這篇pytorch標簽轉(zhuǎn)onehot形式實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python結(jié)合ImageMagick實現(xiàn)多張圖片合并為一個pdf文件的方法
這篇文章主要介紹了Python結(jié)合ImageMagick實現(xiàn)多張圖片合并為一個pdf文件的方法,結(jié)合實例形式分析了Python將圖片文件轉(zhuǎn)換為pdf文件的相關(guān)操作技巧,需要的朋友可以參考下2018-04-04
詳解用Pytest+Allure生成漂亮的HTML圖形化測試報告
這篇文章主要介紹了詳解用Pytest+Allure生成漂亮的HTML圖形化測試報告，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-03-03
今天小編就為大家分享一篇python按照多個條件排序的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-02-02
這篇文章主要介紹了python求素數(shù)示例,打印出素數(shù)列表,需要的朋友可以參考下2014-02-02
django 刪除數(shù)據(jù)庫表后重新同步的方法
今天小編就為大家分享一篇django 刪除數(shù)據(jù)庫表后重新同步的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
這篇文章主要介紹了python對驗證碼降噪的實現(xiàn)示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-11-11
python之np.argmax()及對axis=0或者1的理解
這篇文章主要介紹了python之np.argmax()及對axis=0或者1的理解，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-06-06
這篇文章主要介紹了Python decimal模塊使用方法詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-06-06
這篇文章主要給大家詳細介紹一下Python split()函數(shù)的使用方法，文中有詳細的代碼示例供大家參考，具有一定的參考價值,需要的朋友可以參考下2023-07-07
今天小編就為大家分享一篇解決Pytorch訓練過程中l(wèi)oss不下降的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01