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设计】。
fileinput模块可以对一个或多个文件中的内容进行迭代、遍历等操作。 该模块的input()函数有点类似文件readlines()方法，区别在于: 前者是一个迭代对象，即每次只生成一...
是最初的几个爬虫，让我认识了Python这个新朋友，虽然才刚认识了几天，但感觉有种莫名的默契感。每当在别的地方找不到思路，总能在Python找到解决的办法。自动关机，在平时下载大文件，以...
本文实例讲述了Python学习笔记之自定义函数用法。分享给大家供大家参考，具体如下： 函数能提高应用的模块性，和代码的重复利用率。Python提供了许多内建函数，比如print()等。也...
点乘 import torch x = torch.tensor([[3,3],[3,3]]) y = x*x #x.dot(x) z = torch.mul(x,x) #x.mul...
自适应线性神经网络Adaptive linear network， 是神经网络的入门级别网络。 相对于感知器，采用了f（z）=z的激活函数，属于连续函数。 代价函数为LMS函数，最小均...