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设计】。
这是博主最近一家大公司的面试题，写一个装饰器，限制函数每10s调用一次。当时是笔试的，只写了大概的代码，回来后温习了python装饰器的基础知识，把代码写完了。决定写篇博客记录下。 装饰...
本文实例讲述了Python实现的根据文件名查找数据文件功能。分享给大家供大家参考，具体如下： #-*- coding: UTF-8 -*- import os import shut...
在python中，使用iter函数可以获得有序聚合类型的迭代器，我个人将迭代器理解为带有next指针的单向链表，获取到的迭代器为链表的表头，表头内容为空，next指针指向有序聚合类型的第...
excel 文件内容如下: 读取excel内容: import xlrd from datetime import datetime from xlrd import xldat...
本文实例讲述了Python中DJANGO简单测试的用法。分享给大家供大家参考。具体如下： 这里以facebook台湾的测试版为例。 仅仅测试用户登录，主要说明测试的使用和django环境...