Tensorflow One Hot Encoding Code