Tensorflow One-Hot 函数样例(Python3)

最新推荐文章于 2022-06-02 15:27:35 发布

原创

最新推荐文章于 2022-06-02 15:27:35 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

这篇博客展示了如何在Python3中使用TensorFlow实现One-Hot编码。通过代码示例，解释了如何创建One-Hot向量并进行维度拆分。还提供了一个独特的处理方法作为附录。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Py3代码：

# one-hot 函数的样例
import tensorflow as tf

label = tf.placeholder(tf.int32,[None])
# 直接把 输入的序列进行One-Hot的结果
one_hot = tf.one_hot(label, 3, 1, 0)
# 进行转置
one_hot_new = tf.transpose(one_hot, perm=[1,0])
one_hot_new = tf.cast(one_hot_new, tf.float32)
# one_hot_new[2] = one_hot_new[2] * 1.5

# 按照每两维的大小进行拆分
one_hot_new_1 = tf.dynamic_partition(one_hot_new, [0, 0, 1],