tf.one_hot()进行独热编码