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标签就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。
说明 在python中，if作为条件语句，当if后面的条件参数为真时，则执行后面的语句块，反之跳过，为了深入理解if语句，我们需要知道if语句的真假判断方式。 示例 在python交互器...
前言： 在做一个商城项目的时候，需要实现商品搜索功能。 说到搜索，第一时间想到的是数据库的 select * from tb_sku where name like %苹果手机% 或者...
由于近日写工作汇报 交一份概要和 excel版本 发现手写 太麻烦 想写个代码来自动 读取一个文本中的所有内容 按需求写入 对应excel和概要中 自动完成 就找了找python 的类库...
使用pythonbrew 复制代码 代码如下: easy_install pythonbrew [root@li637-23 schirm]# pythonbrew_install We...
%还是format 1、皇城PK Python中格式化字符串目前有两种阵营：%和format，我们应该选择哪种呢？ 自从Python2.6引入了format这个格式化字符串的方法之后，我...