tf.argmax用法

import tensorflow as tf
c= [[[16,6,3],[4,6,5]], 
    [[8,7,9], [11,12,10]]]
with tf.Session() as sess:
    print(sess.run(tf.argmax(c,0)))
    print('---------------------')
    print(sess.run(tf.argmax(c,1)))
    print('---------------------')
    print(sess.run(tf.argmax(c,2)))



[[0 1 1]
 [1 1 1]]
---------------------
[[0 0 1]
 [1 1 1]]
---------------------
[[0 1]
 [2 1]]

(C,0) 在矩阵第0轴找最大值的索引，16和8比，6和7比，3和9比，4和11比 。。。

[[16,6,3],[4,6,5]], 
 [[8,7,9], [11,12,10]]

(C,1) 在矩阵第1轴找最大值的索引, 16和4比，6和6比，3和5比。。。

[[16,6,3],
[4,6,5]], 

 [[8,7,9],
  [11,12,10]]

(C,2) 在矩阵第2轴找最大值的索引，每行進行比較

[16,6,3],
[4,6,5], 

 [8,7,9], 
 [11,12,10]