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可变参数*args和**kwargs用法。分享给大家供大家参考，具体如下： 一句话简单概括：当函数的参数不确定的时候就需要用到*args和**kwargs，前者...
本文实例讲述了Python编程实现生成特定范围内不重复多个随机数的2种方法。分享给大家供大家参考，具体如下： 在近期进行的一个实验中，需要将数据按一定比例随机分割为两个部分。这一问题的核...
刚刚学习使用phantomjs，根据网上帖子自己手动改了一个延时截图功能，发现延时功能就是不能执行，最后一点点排查出了问题。 看代码： var page = require('web...
前言 本篇和大家分享的是使用python简化对jar包操作命令，封装成简短关键字或词，达到操作简便的目的。最近在回顾和构思shell脚本工具，后面一些文章应该会分享shell内容，希望大...
在“循环”一节，我们已经讨论了Python基本的循环语法。这一节，我们将接触更加灵活的循环方式。 range() 在Python中，for循环后的in跟随一个序列的话，循环每次使用的序列...