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 地图经纬度转换、纠偏的代码如下所示： # -*- coding: utf-8 -*- import json import urllib import math x_p...
本文实例为大家分享了用matplotlib中scatter方法画散点图的具体代码，供大家参考，具体内容如下 1、最简单的绘制方式 绘制散点图是数据分析过程中的常见需求。python中最有...
layui的table的自定义模板需要用到{{}}，但是和Django的{{}}冲突了，layui的{{}}失效了 解决方法：从Django 1.5开始，支持{% verbatim %}...
一、简单使用 复制代码 代码如下: def TestLogBasic(): import logging l...
业务需求 需要将文本文件分每两行一组 jb51.txt 1:www.jb51.net 2:www.jb51.net 3:www.jb51.net 4:www.jb51.net 5:ww...