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设计】。
使用Matplotlib绘制的图表的默认坐标轴是在左下角的，这样对于一些函数的显示不是非常方便，要改变坐标轴的默认显示方式主要要使用gca()方法 plt.gca()表示 Get cur...
本文实例讲述了Python统计一个字符串中每个字符出现了多少次的方法。分享给大家供大家参考，具体如下： #coding=utf-8 #统计一个字符串中的每一个字符出现了多少次 #定义...
我就废话不多说了，直接上代码吧！ # -*- coding: utf-8 -*- """ Created on Thu Jun 22 17:03:16 2017 @author: y...
python微信跳一跳，前言 这是python玩跳一跳系列博文中一篇，主要内容是用颜色识别的方法来进行跳跳小人的定位。 颜色识别 通过观察，我们可以发现，尽管背景和棋子在不停的变化，但...
我就废话不多说了，直接上代码吧！ import pandas as pd import numpy as np import matplotlib.pyplot as plt fig...