tf.one_hot

Learn tf.one_hot Step by Step