index = [0, 1, 2, 3]
one_hot = tf.one_hot(index, 5)

输出 

array([[1., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0.],
       [0., 0., 1., 0., 0.],
       [0., 0., 0., 1., 0.]], dtype=float32)