In TensorFlow, what is the argument 'axis' in the function 'tf.one_hot' In TensorFlow, what is the argument 'axis' in the function 'tf.one_hot' python-3.x python-3.x