tf.argmax()函数中axis参数理解

tf.argmax是tensorflow用numpy的np.argmax实现的，它能给出某个tensor对象在某一维上的其数据最大值所在的索引值，常用于metric（如acc）的计算

tf.argmax()函数中有个axis参数（轴），该参数能指定按照哪个维度计算。
如 在矩阵的结构中，axis可被设置为0或1，分别表示
0：按列计算，1：行计算

下面是具体的示例，便于直观理解（因tf.argmax()的axis的用法与numpy中的argmax相同，为方便起见，本文使用了numpy.argmax()用于示例）
1.

当axis=0时，以列为单位，第一列的最大值为索引为3的8，所以第一个值为3，以此类推

2. axis=1

---

当axis=1时，以行为单位，第一行的最大值为索引为2的3，所以第一个结果为2