TensorFlow函数（七）tf.argmax()

tf.argmax(input, dimension, name=None) 参数： input：输入数据 dimension：按某维度查找。 　　　　dimension=0：按列 ...