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