网上彩票购买

码迷,bjsfhd.cn
网上彩票购买 > 其他好文 > 详细

pytorch中onehot编码转为普通label标签

时间:2019-06-30 00:19:47      阅读:16      评论:0      收藏:0      [点我收藏+]

标签:没有   code   编码   squeeze   自己   tensor   hot   pytorch   ORC   

label转onehot的很多,但是onehot转label的有点难找,所以就只能自己实现以下,用的topk函数,不知道有没有更好的实现

one_hot = torch.tensor([[0,0,1],[0,1,0],[0,1,0]])
print(one_hot)
label = torch.topk(one_hot, 1)[1].squeeze(1)
print(label)

 

tensor([[0, 0, 1],
[0, 1, 0],
[0, 1, 0]])
tensor([2, 1, 1])

pytorch中onehot编码转为普通label标签

标签:没有   code   编码   squeeze   自己   tensor   hot   pytorch   ORC   

原文地址:https://www.cnblogs.com/youmuchen/p/11108247.html

(0)
(0)
   
举报
评论 一句话评论(0
0条  
登录后才能评论!
           
© 2014 bjsfhd.cn 版权所有 京ICP备13008772号-2
迷上了代码!
手机彩票app 手机彩票app 彩票开奖查询 网上彩票购买 手机彩票app 秒速彩票投注 网上彩票购买 网上彩票购买 彩票app 手机彩票app