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设计】。
第一步：字母转数字 英文字母转对应数字相对简单，可以在命令行输入一行需要转换的英文字母，然后对每一个字母在整个字母表中匹配，并返回相应的位数，然后累加这些位数即可。过程中，为了使结果更...
前言 本文紧接着前一篇的入门教程，会介绍一些关于pandas的进阶知识。建议读者在阅读本文之前先看完pandas入门教程。 同样的，本文的测试数据和源码可以在这里获取： Github：p...
在django项目中, 一个工程中存在多个APP应用很常见. 有时候希望不同的APP连接不同的数据库，这个时候需要建立多个数据库连接。 1. 修改项目的 settings 配置 在 s...
如下所示： # -*- coding: utf-8 -*- # 要求：输入一行字符，分别统计出其中英文字母、空格、数字和其它字符的个数。 def count(s): count_a...
在计算机高级中语言，运算符是比较多样化的。其实，也都源于我们日常的需要。 算术运算符 前面已经讲过了四则运算，其中涉及到一些运算符：加减乘除，对应的符号分别是：+ - * /，此外，还有...