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设计】。
本文实例讲述了Python设计模式之工厂方法模式。分享给大家供大家参考，具体如下： 工厂方法模式(Factory Method Pattern):定义一个用于创建对象的接口,让子类决定实...
在之前的有关线程，进程的博客中，我们介绍了它们各自在同一个程序中的通信方法。但是不同程序，甚至不同编程语言所写的应用软件之间的通信，以前所介绍的线程、进程队列便不再适用了；此种情况便只能...
一、设置开启SMTP服务并获取授权码 0.如果使用第三方邮件服务器SMTP服务来发送邮件，首先要在邮箱设置里面开启POP3/SMTP/IMAP服务，下面以163邮箱为例，其它邮箱设置方法...
本文实例讲述了Python实现模拟登录及表单提交的方法。分享给大家供大家参考。具体实现方法如下： # -*- coding: utf-8 -*- import re import...
如下所示： import urllib,json,requests url = 'http://127.0.0.1:8000/account/login' headers = {}...