从单热编码转换为类标签

Question

我有一个名为y的张量，它具有来自类标签的单热编码的值： y = [[0. 0. 1. ...，0. 0。0。]，[1. 0. 0. ...，0. 0。0。]，[0。0. 0 .. 。，0.1.1。]， ...，[0。0. 0. ...，0.0。0。]，[0. 0. 0 ...，0. 0。1。]，[0。0.1。 ..，0. 0. 0.]] 所以这里第一行有第三个元素为'1'所以它代表类标签 对于那个图像。 我试图从给定的一个热编码数组中获取所有类标签， 或者给定的例子它应该是这样的： y = [2,0,8，...，9,2]

Answer 1

我认为最简单的方法是：

import numpy as np
y = np.argmax(y)