depth: Skalar yang mendefinisikan kedalaman satu dimensi panas.
indeks: Tensor indeks.
Ini contoh yang diberikan dalam dokumentasi tensorflow.
1. Hanya Menentukan indeks dan kedalaman (Nilai Default on_value adalah 1 dan off_value adalah 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.]]
- Menentukan on_value dan off_value
indeks = [0, 2, -1, 1]
kedalaman = 3
tf.one_hot (indeks, kedalaman, on_value = 5.0, off_value = 0.0, sumbu = -1)
Keluaran ##: [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)
Anda juga dapat melihat kode di GitHub