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