tf.one_hot(indices,#输入,这里是一维的depth,# one hotdimension.on_value=None,#output 默认1off_value=None,#output 默认0axis=None,#根据我的实验,默认为1dtype=None,name=None)
测试程序,一般说,有几类,depth等于分类的类别数
有10类,则depth=10
import tensorflow as tf
import numpy as np
z=np.array([0,1,3,4,6,4,7,1,0])
y=tf.one_hot(z,10)
with tf.Session()as sess:print(z)print(sess.run(y))
输出: