代码之家 › 专栏 › 技术社区 › KOB

创建张量,其中给定索引的所有元素均为1,其余元素均为0

one-hot-encoding tensor mask tensorflow

KOB · 技术社区 · 8 年前

我有一个占位符 lengths = tf.placeholder(tf.int32, [10]) . 分配给此占位符的10个值中的每一个都是<=我现在想创建一个二维张量,叫做 masks ,其形状为[10,25],其中长度为25的10个向量中的每一个都具有第一个 n 元素设置为 1 ,其余设置为 0 -与 n 是中的对应值 lengths

例如:

lengths = [4, 6, 7, ...]

-> masks = [[1, 1, 1, 1, 0, 0, 0, 0, ..., 0],
            [1, 1, 1, 1, 1, 1, 0, 0, ..., 0],
            [1, 1, 1, 1, 1, 1, 1, 0, ..., 0],
            ...
           ]

1 回复 | 直到 8 年前

Engineero 8 年前

您可以将长度重塑为(10,1)张量,然后将其与其他序列/索引进行比较 0,1,2,3,...,25 ,由于广播将导致如果指数小于长度,则为假 ; 然后可以将布尔结果转换为 1 和 0 :

lengths = tf.constant([4, 6, 7])
n_features = 25
â
import tensorflow as tf
â
masks = tf.cast(tf.range(n_features) < tf.reshape(lengths, (-1, 1)), tf.int8)

with tf.Session() as sess:
    print(sess.run(masks))

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

推荐文章

2D_ · 使用pandas或scikit learn在多维数组上进行一次热编码

8 年前

Walter U. · “特征重要性”的“一个热编码”变量的显示名称

8 年前

under_the_sea_salad · 我应该使用哪个Spark ML特征转换器将一列短语转换为固定长度的向量?

8 年前

Raag Agrawal · H2o GLM仅与某些预测因子相互作用

8 年前

Kathiravan Natarajan · 将Tensorflow预测导出到csv,但结果包含所有零-这是因为一个热端吗?

8 年前

moobi · 在训练中使用一个热编码后,获得用于回归模型预测的数据点的正确形状

8 年前

DVL · 熊猫:如何在不寻常的文本顺序中分解

8 年前