tensorflow对tf.argmax()的理解

在tensorflow中，axis=1对列进行操作，axis=0对行进行操作

• 例如，argmax中axis=1就是对(每行中)不同列元素进行比较，得到最大值。

a = [[0, 0, 0, 1, 0, 0],
 [0, 0, 1, 0, 0, 0],
 [0, 1, 0, 0, 0, 0],
 [0, 0, 0, 0, 0, 1]]
 
 # axis=1, 每行的不同列进行比较，取对应最大值的索引
tf.argmax(a, 1) => [3 2 1 5]

# axis=０, 每列的不同行进行比较，去对应最大值的索引
tf.argmax(a, 0) => [0 2 1 0 0 3]