从one-hot编码到类标签的转换

Question

我有一个名为y的张量，它的值来自类标签上的one-hot编码:y=[ 0。0。1.，0.0。0.，1.0。0。...，0.0。0。，0。0。0。...，0.1. 0.，...，0.0。0。...，0.0。0。，0。0。0。...，0.0。1.，0.0。1.，0.0。0。]因此，这里第一行的第三个元素是'1‘，所以它表示该图像的类标签。我试图从给定的one-hot编码数组中获取所有类标签，或者给定的示例应该是这样的:y= 2,0,8，...,9,2

Answer 1

我认为最简单的方法是：

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