tf.one_hot(indices, depth, on_value, off_value, axis)
- indices是一个列表,指定张量中独热向量的独热位置,或者说indeces是非负整数表示的标签列表。len(indices)就是分类的类别数。
- tf.one_hot返回的张量的阶数为indeces的阶数+1。
- 当indices的某个分量取-1时,即对应的向量没有独热值。
-
depth是每个独热向量的维度
-
on_value是独热值
-
off_value是非独热值
-
axis指定第几阶为depth维独热向量,默认为-1,即,指定张量的最后一维为独热向量