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中的urllib模块用来处理url相关的操作，unquote方法对应javascript中的urldecode方法，它对url进行解码，把类似"%xx"的字符替换成...
当我们的函数接收参数为任意个，或者不能确定参数个数时，我们，可以利用 * 来定义任意数目的参数，这个函数调用时，其所有不匹配的位置参数会被赋值为元组，我们可以在函数利用循环或索...
如下所示： ''' Created on 2018-4-20 例子:每天凌晨3点执行func方法 ''' import datetime import threading def...
Python实现插入排序的一般范例为： #coding=cp936 #coding=cp936 #插入排序算法 def InsertionSort(A): for j in r...
用pytorch训练一个神经网络时，我们通常会很关心模型的参数总量。下面分别介绍来两种方法求模型参数 一 .求得每一层的模型参数，然后自然的可以计算出总的参数。 1.先初始化一个网络模型...