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设计】。
搭建FTP，或者是搭建网络文件系统，这些方法都能够实现Linux的目录共享。但是FTP和网络文件系统的功能都过于强大，因此它们都有一些不够方便的地方。比如你想快速共享Linux系统的...
今晚终于解决了串口发送的问题，更改代码如下： def write(self, data): if self.alive: if self.serSer.isOpe...
在python3的sorted中去掉了cmp参数，转而推荐“key+lambda”的方式来排序。 如果需要对python的list进行多级排序。有如下的数据： list_num =...
这篇文章主要介绍了python解析命令行参数的三种方法详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 python解析命令行参数主...
本文实例讲述了Python随机数用法。分享给大家供大家参考，具体如下： 1. random.seed(int) 给随机数对象一个种子值，用于产生随机序列。 对于同一个种子值的输入，之后产...