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设计】。
复制代码 代码如下:import time,datetime import urllib2 def chk_qq(qqnum)...
复制代码 代码如下:import sysfrom PyQt4 import QtGui, QtCoreclass Window( QtGui.QMainWindow): &nb...
本文实例讲述了python检测某个变量是否有定义的方法。分享给大家供大家参考。具体如下： 第一种方法使用内置函数locals()： 'testvar' in&nb...
python继承，python丰富的类因为继承而变得多姿多彩，如果语言不支持继承，那么类就没什么优势。 1、首先我们来定义两个类 一个dog类，一个bird类class Dog: &nb...
本文实例讲述了Python打开文件、文件读写操作、with方式、文件常用函数。分享给大家供大家参考，具体如下： 打开文件： 在python3中，打开文件的函数是: open(file,...