tf.one_hot

tf.one_hot