tf.argmax

tf.argmax（inputs，axis）

根据axis取值不同，返回每行或者每列最大值的索引。axis=0，则比较每列最大元素所在的索引记录下来。axis=1，则是每行最大索引记录下。

test = np.array([
[1, 2, 3],
 [2, 3, 4], 
 [5, 4, 3], 
 [8, 7, 2]])
np.argmax(test, 0)　　　＃输出：array([3, 3, 1]
np.argmax(test, 1)　　　＃输出：array([2, 2, 0, 0]

注意！！！返回的是数的位置，而不是数的值