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實(shí)現(xiàn)RGB與HSI顏色空間的互換方式
今天小編就為大家分享一篇Python實(shí)現(xiàn)RGB與HSI顏色空間的互換方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-11-11
這篇文章主要為大家詳細(xì)介紹了MySQL適配器PyMySQL的相關(guān)資料，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2017-09-09
在安裝Anaconda之前,有的已經(jīng)安裝過(guò)一個(gè)Python版本了,但是又不想刪除這個(gè)Python版本,該怎么辦呢??這篇文章主要給大家介紹了關(guān)于保留已有python安裝Anaconda的幾種法推薦,需要的朋友可以參考下2023-12-12
深度學(xué)習(xí)的MNIST手寫(xiě)數(shù)字?jǐn)?shù)據(jù)集識(shí)別方式(準(zhǔn)確率99%,附代碼)
這篇文章主要介紹了深度學(xué)習(xí)的MNIST手寫(xiě)數(shù)字?jǐn)?shù)據(jù)集識(shí)別方式(準(zhǔn)確率99%,附代碼),具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-06-06
Python實(shí)現(xiàn)截圖生成符合markdown的鏈接
之前是用的是typora來(lái)寫(xiě)的文章，最近typora最近開(kāi)始收費(fèi)了，所以就不想用了，于是找到了一個(gè)替代品MarkText。本文將介紹如何通過(guò)Python實(shí)現(xiàn)截圖自動(dòng)生成符合markdown的鏈接，感興趣的可以了解一下2022-01-01
這篇文章主要介紹了Python映射拆分操作符用法,實(shí)例分析了Python映射拆分操作符**的使用技巧,需要的朋友可以參考下2015-05-05
Python?pywin32實(shí)現(xiàn)word與Excel的處理
這篇文章主要介紹了Python?pywin32實(shí)現(xiàn)word與Excel的處理，pywin32處理Word大多數(shù)用于格式轉(zhuǎn)換，因?yàn)橐话阕x寫(xiě)操作都可以借助python-docx實(shí)現(xiàn)，除非真的有特殊要求，但大部分企業(yè)對(duì)Wrod操作不會(huì)有太多復(fù)雜需求2022-08-08
python基于機(jī)器學(xué)習(xí)預(yù)測(cè)股票交易信號(hào)
近年來(lái)，隨著技術(shù)的發(fā)展，機(jī)器學(xué)習(xí)和深度學(xué)習(xí)在金融資產(chǎn)量化研究上的應(yīng)用越來(lái)越廣泛和深入。目前，大量數(shù)據(jù)科學(xué)家在Kaggle網(wǎng)站上發(fā)布了使用機(jī)器學(xué)習(xí)/深度學(xué)習(xí)模型對(duì)股票、期貨、比特幣等金融資產(chǎn)做預(yù)測(cè)和分析的文章。本文就來(lái)看看如何用python預(yù)測(cè)股票交易信號(hào)2021-05-05
python解壓縮文件或文件夾兩種常見(jiàn)方式(附代碼)
這篇文章主要介紹了python解壓縮文件或文件夾兩種常見(jiàn)方式,Python的zipfile和shutil模塊提供了強(qiáng)大的文件和文件夾壓縮與解壓縮功能,zipfile模塊適合精細(xì)控制,而shutil模塊則快速簡(jiǎn)單,文中提供了詳細(xì)的代碼示例,需要的朋友可以參考下2025-04-04
這不是一篇關(guān)于Python異常的全面介紹的文章，這只是在學(xué)習(xí)Python異常后的一篇筆記式的記錄和小結(jié)性質(zhì)的文章2013-10-10