# 首先看傳入的tensor的size，傳入的tensor的維度不夠則需要添加維度 # 這里的labels的size是tensor([n])的，傳入需要添加維度 # 如果出現上述問題，只需要在后面加上to.(torch.int64)即可解決 # n是要分的種類
labels = torch.nn.functional.one_hot(labels.unsqueeze(0).to(torch.int64), n)
親測有效！！！
RuntimeError: one_hot is only applicable to index tensor
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
pytorch 獨熱編碼報錯的解決辦法：one_hot is only applicable to index tensor
什么是One_Hot?
sklearn one_hot 操作
『TensorFlow』one_hot化標簽
Tensorflow中one_hot() 函數用法
深度學習中將類別標簽映射到one_hot向量
one-hot 編碼
one-hot編碼
one-hot編碼理解
深度學習原理與框架-Tensorflow卷積神經網絡-cifar10圖片分類(代碼) 1.tf.nn.lrn(局部響應歸一化操作) 2.random.sample(在列表中隨機選值) 3.tf.one_hot(對標簽進行one_hot編碼)