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

您可以将长度重塑为(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]]