Tensorflow One Hot Encoding Python