tensorflow--tf.one_hot()函数示例

最新推荐文章于 2024-07-25 17:32:42 发布

哎呦不错的温jay

最新推荐文章于 2024-07-25 17:32:42 发布

阅读量2.8w

点赞数 8

分类专栏： tensorflow 文章标签： tensorflow python

本文链接：https://blog.csdn.net/wenqiwenqi123/article/details/78055740

版权

tensorflow 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

tensorflow中tf.one_hot()函数的作用是将一个值化为一个概率分布的向量，一般用于分类问题。

具体用法以及作用见以下代码：

import numpy as np
import tensorflow as tf

SIZE=6
CLASS=8
label1=tf.constant([0,1,2,3,4,5,6,7])
sess1=tf.Session()
print('label1:',sess1.run(label1))
b = tf.one_hot(label1,CLASS,1,0)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    sess.run(b)
    print('after one_hot',sess.run(b))

最后的输出为：

label1: [0 1 2 3 4 5 6 7]
after one_hot:

[[1 0 0 0 0 0 0 0]
[0 1 0 0 0 0 0 0]
[0 0 1 0 0 0 0 0]
[0 0 0 1 0 0 0 0]
[0 0 0 0 1 0 0 0]
[0 0 0 0 0 1 0 0]
[0 0 0 0 0 0 1 0]
[0 0 0 0 0 0 0 1]]