Tensorflow One Hot Encoder