Tensorflow使用0和1初始化列表，類似於tf.one_hot

Question

我目前有一個數值列表，其值爲[1, 3, 5, 7, 3]。目前，我正在使用tf.one_hot爲列表中的每個值獲取一個熱矢量/張量代表。

1 = [0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0] 
3 = [0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0] 
5 = [0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0] 
7 = [0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0] 
3 = [0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0] 

在Tensorflow有什麼辦法/功能，讓我做同樣的事情，但初始化爲0的所有值與1秒的價值？

所需的結果：

1 = [1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0] 
3 = [1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0] 
5 = [1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0] 
7 = [1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0] 
3 = [1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0] 

Answer 1

您正在尋找tf.sequence_mask。

TensorFlow API