tf.one_hot 实例

Examples
=========

Suppose that


indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1


Then output is [4 x 3]:


output =
[5.0 0.0 0.0]
[0.0 0.0 5.0]
[0.0 0.0 0.0]
[0.0 5.0 0.0]


Suppose that


indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1


Then output is [2 x 2 x 3]:


output =
[
[1.0, 0.0, 0.0]
[0.0, 0.0, 1.0]
][
[0.0, 1.0, 0.0]
[0.0, 0.0, 0.0]
]


Using default values for on_value and off_value:


indices = [0, 1, 2]
depth = 3


The output will be


output =
[[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]]