tensorflow one-hot encoding