TF中的One-hot

• tf.one_hot (待转换数据, depth=几分类)

import tensorflow as tf

classes = 3
labels = tf.constant([1, 0, 2])
output = tf.one_hot(labels, classes)
print(output)

注: 当列表中的变量带有小数的时候,one-hot报错