You can use label encoder or one-hot encoder to represent your label.
Example for one-hot encoding,
indices = [0, 1, 2]
depth = 3
tf.one_hot(indices, depth)  
# output: [3 x 3]# [[1., 0., 0.],#  [0., 1., 0.],#  [0., 0., 1.]]