TensorFlow函數（七）tf.argmax()

tf.argmax(input, dimension, name=None) 參數： input：輸入數據 dimension：按某維度查找。 　　　　dimension=0：按列 ...