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设计】。
在训练循环中，定期调用 saver.save() 方法，向文件夹中写入包含了当前模型中所有可训练变量的 checkpoint 文件。 saver.save(sess, FLAGS.t...
本文实例讲述了python django下载大的csv文件实现方法。分享给大家供大家参考，具体如下： 接手他人项目，第一个要优化的点是导出csv的功能，而且要支持比较多的数据导出，以前用...
1、下载安装好PyCharm 专业版后打开或者新建一个Python项目，找到View导航栏， 如下图： 在Tool Windows下可以找到SciView按钮，但是每次打开PyChar...
首先先说一下python2与python3的选择 许多刚入门 Python 的朋友都在纠结的的问题是：我应该选择学习 python2 还是 python3？ 对此，回答是：果断 Pyth...
一、枚举与字典类型 字典类型的缺点： 1.值可变 2.没有防止相同标签的功能 枚举的特点： 1.枚举类的值不可以被外界更改 2.不能存在相同的标签，但允许不同标签存在相同的枚举值，...