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设计】。
TensorFlow™是一个基于数据流编程（dataflow programming）的符号数学系统，被广泛应用于各类机器学习（machine learning）算法的编程实...
一、TensorFlow常规模型加载方法 保存模型 tf.train.Saver()类，.save(sess, ckpt文件目录)方法 参数名称...
__import__() 函数用于动态加载类和函数 。 如果一个模块经常变化就可以使用 __import__() 来动态载入。 语法 __import__ 语法： __import...
分形几何学的基本思想：客观事物具有自相似性的层次结构，局部和整体在形态，功能，信息，时间，空间等方面具有统计意义上的相似性，称为自相似性，自相似性是指局部是整体成比例缩小的性质。 我们先...
本文实例讲述了Flask框架学习笔记之模板操作。分享给大家供大家参考，具体如下： flask的模板引擎是Jinja2。 引入模板的好处是增加程序的可读性和易维护性，从而不用将一堆html...