depth: un scalaire définissant la profondeur de la seule dimension chaude.
indices: un tenseur d'indices.
C'est l'exemple donné dans la documentation tensorflow.
1. Spécifier uniquement les indices et la profondeur (les valeurs par défaut de on_value est 1 et off_value est 0)
indices = [0, 1, 2]
depth = 3
tf.one_hot(indices, depth) # output: [3 x 3]
# [[1., 0., 0.],
# [0., 1., 0.],
# [0., 0., 1.]]
- Spécification de on_value et off_value
indices = [0, 2, -1, 1]
profondeur = 3
tf.one_hot (indices, profondeur, on_value = 5.0, off_value = 0.0, axis = -1)
## sortie: [4 x 3]
# [[5.0, 0.0, 0.0], # one_hot (0)
# [0.0, 0.0, 5.0], # one_hot (2)
# [0,0, 0,0, 0,0], # one_hot (-1)
# [0.0, 5.0, 0.0]] # one_hot (1)
Vous pouvez également voir le code sur GitHub