Convert label index to one-hot form in tensorflow

 
 
import tensorflow as tf
index=[0,1,2,3]
one_hot=tf.one_hot(a,5)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())

    print(sess.run(one_hot))



Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=326069391&siteId=291194637