tf.one_hot函数用法

最新推荐文章于 2023-01-13 11:53:22 发布

星尘亦星辰

最新推荐文章于 2023-01-13 11:53:22 发布

阅读量949

点赞数 1

分类专栏：深度学习（tensorflow2.0）文章标签： tensoflow2.0

本文链接：https://blog.csdn.net/Mrweng1996/article/details/103254093

版权

深度学习（tensorflow2.0）专栏收录该内容

15 篇文章 0 订阅

订阅专栏

具体用法如下：

In [4]: y = tf.constant([1,2,3,0,2])

In [5]: y = tf.one_hot(y,depth=4)

In [7]: y
Out[7]:
<tf.Tensor: id=6, shape=(5, 4), dtype=float32, numpy=
array([[0., 1., 0., 0.],
       [0., 0., 1., 0.],
       [0., 0., 0., 1.],
       [1., 0., 0., 0.],
       [0., 0., 1., 0.]], dtype=float32)>