Tensorflow Initialiser la liste avec 0s et 1s similaires à tf.one_hot

Question

J'ai actuellement une liste de valeurs sur une séquence de temps dire que les valeurs sont [1, 3, 5, 7, 3]. Actuellement, j'utilise tf.one_hot pour obtenir un représentant vectoriel/tensoriel chaud pour chaque valeur de la liste.

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] 

En tensorflow est-il un moyen/fonction qui me permettrait de faire quelque chose de similaire, mais toutes les valeurs de l'initialise 0 à la valeur avec 1s?

Résultat souhaité:

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

Vous cherchez tf.sequence_mask.

TensorFlow API