Tensorflow One Hot Example