Tensorflow One Hot Encoding