tf.one_hot()

TensorFlow1.2函数原型

https://www.tensorflow.org/versions/r1.2/api_docs/python/tf/one_hot

tf.one_hot
one_hot(
    indices,
    depth, #标量，指定了每个one-hot的dimension
    on_value=None, #默认为1
    off_value=None, #默认为0
    axis=None,  #默认为-1
    dtype=None, #当on_value, off_value没有指定时,默认为tf.float32
    name=None
)

数据间没有序关系的时候可以用one_hot编码。

indices所代表的位置，用on_value替代，其余位置用off_value替代。

输出维度：

1. 当indices是一个标量，则输出的shape为长度为depth的向量；

tf.reset_default_graph()

indices = 1
depth = 6

one_hot = tf.one_hot(indices=indices,
                     depth=depth,