Tensorflow 创建一个张量,其中给定索引下的所有元素都是1,其余的都是0

Tensorflow 创建一个张量,其中给定索引下的所有元素都是1,其余的都是0,tensorflow,mask,tensor,one-hot-encoding,Tensorflow,Mask,Tensor,One Hot Encoding,我有一个占位符length=tf.placeholder(tf.int32[10])。分配给该占位符的10个值中的每一个都是 您可以将长度重塑为(10,1)张量,然后将其与另一个序列/索引进行比较0,1,2,3,…,25,由于广播,如果索引小于长度,则结果为真,否则为假;然后可以将布尔结果强制转换为1和0: length=tf.常数([4,6,7]) n_特征=25 ​ 导入tensorflow作为tf ​ 掩模=tf.cast(tf.range(n_特征)

我有一个占位符
length=tf.placeholder(tf.int32[10])
。分配给该占位符的10个值中的每一个都是 您可以将长度重塑为(10,1)张量,然后将其与另一个序列/索引进行比较
0,1,2,3,…,25
,由于广播,如果索引小于长度,则结果为真,否则为假;然后可以将布尔结果强制转换为
1
0

length=tf.常数([4,6,7])
n_特征=25
​
导入tensorflow作为tf
​
掩模=tf.cast(tf.range(n_特征)