tensorflow2.X转换Onehot

编程语言 > 机器学习 tensorflow python

发布时间 : 2020-07-14 00:00

字数:21 阅读 :

import tensorflow as tf 


x           = tf.constant([1,5,9,0])
x_onehot    = tf.one_hot(x,depth=10)


print(x)
print(x_onehot)

转载请注明来源 https://tianweiye.github.io