tf.argmax()解析

tf.argmax()函數其實就是求最大數值所在下標。只是參數不同的時候，表達的列還是行的最大數值。

如tf.argmax(array,0)表示的是求列最大值的索引下標。

   tf.argmax(array,1)則表示求行最大值的索引下標。

例子：

array([[[ 1,  2,  3],
        [ 4,  3,  2],
        [ 2,  2,  5]]

tf.argmax(array,0)  #輸出的是列最大值的下標[1,1,2]

tf.argmax(array,1)  #輸出的是行最大值的下標[2,0,2]