pytorch实现onehot编码转为普通label标签
作者:swuxyj 时间:2023-04-18 10:19:21
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])
来源:https://www.cnblogs.com/youmuchen/p/11108247.html
标签:pytorch,onehot,label
0
投稿
猜你喜欢
PHP高级编程实例:编写守护进程
2023-10-27 02:03:22
python利用opencv实现SIFT特征提取与匹配
2021-07-22 19:08:46
python根据用户需求输入想爬取的内容及页数爬取图片方法详解
2022-12-21 16:31:19
Python 时间戳之获取整点凌晨时间戳的操作方法
2023-10-10 19:42:33
python3 http提交json参数并获取返回值的方法
2023-09-26 12:30:06
laravel yajra插件 datatable的使用详解
2023-11-18 20:05:06
ASP调试方法图文教程
2008-01-26 19:32:00
2022年最新Typora的破解方法
2023-08-29 04:36:03
关于Matplotlib绘制动态实时曲线的方法改进指南
2021-03-17 23:48:10
Ajax+Servlet+jsp显示搜索效果
2023-06-14 08:29:34
Web设计中的黄金分割[译]
2009-02-20 13:41:00
linux系统使用python监控apache服务器进程脚本分享
2021-10-15 00:15:41
python爬取淘宝商品详情页数据
2021-05-28 14:38:08
js 将json字符串转换为json对象的方法解析
2023-07-22 21:41:49
Python实现Keras搭建神经网络训练分类模型教程
2023-08-06 12:33:03
JS应用正则表达式转换大小写示例
2024-04-26 17:11:14
python如何每天在指定时间段运行程序及关闭程序
2021-08-05 05:42:05
mysql性能优化脚本mysqltuner.pl使用介绍
2024-01-17 16:37:02
Mysql DateTime 查询问题解析
2024-01-23 06:00:53
python使用opencv对图像添加噪声(高斯/椒盐/泊松/斑点)
2022-01-27 13:05:08