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设计】。
这只是符合比较正常的需求和场景。 #一、适用合并两个字典（key不能相同否则会被覆盖），简单，好用。 A = {'a': 11, 'b': 22} B = {'c': 48, 'd...
前言 大家应该都知道，Centos是目前最为流行的Linux服务器系统，其默认的Python 2.x，但是根据python社区的规划，在不久之后，整个社区将向Python3迁移，且将不在...
本文实例讲述了Python实现嵌套列表及字典并按某一元素去重复功能。分享给大家供大家参考，具体如下： #! /usr/bin/env python #coding=utf-8 cla...
思路 利用scapy库,在这个库的基础下能够做很多的事情,python读取pcap包网上一找一大把 将读取出来的pcap包改一个名字,然后写回,这不就OK了吗 写回的函数是:scapy....
一、图像色彩通道拆分 import cv2 img1 = cv2.imread(r"D:\OpencvTest\example.jpg", cv2.IMREAD_COLOR) #...