import tensorflow as tf
a = tf.one_hot(5, 10)
b = tf.argmax(a)
with tf.Session() as sess:
print(sess.run(a))
print(sess.run(b))
tensorflow之one hot
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
import tensorflow as tf
a = tf.one_hot(5, 10)
b = tf.argmax(a)
with tf.Session() as sess:
print(sess.run(a))
print(sess.run(b))