【发布时间】:2017-07-27 03:35:36
【问题描述】:
我有一个名为 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]
【问题讨论】:
-
您可以使用 tf.argmax(y,1) 在 tensorflow 图中执行此操作
标签: tensorflow