在使用argmax()函数时，比如在深度学习里面计算acc经常要用到这个参数，这个参数返回的是沿轴axis最大值的索引值，对于，tensorflow和numpy用法是一样的，此处我就用numpy来举例说明。

argmax(a, axis=None, out=None)
# a 表示array
# axis 表示指定的轴，默认是None，表示把array平铺，
# out 默认为None，如果指定，那么返回的结果会插入其中
  • 1
  • 2
  • 3
  • 4

对于axis默认为None的情况

首先我们看axis=None，也就是默认的时候：

二维array情况

a = np.array([[2,5,6],[7,6,1]])
print(np.argmax(a))
  • 1
  • 2

输出结果为3，因为a里面7是最大的，如果没有指定axis，默认就是None，相当于把array平铺为：[2,5,6,7,6,1],那么结果就是3，因为索引3对应的值最大，再来看一下三维的情况 
### 三维array的情况

b = np.random.randint(20,size=[3,2,2])
print(b)
[[[ 0 16]
  [14  5]]

 [[16  6]
  [19  2]]

 [[11 11]
  [ 5  7]]]
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10

这个里面，19最大，把它平铺，19对应的索引就是6，那么np.argmax(b)就是6,经过验证输出就是6

对于给定axis的情况

如果给定axis，那么返回的就是沿着给定轴的最大索引，所谓沿着，我觉得就是投影，就是沿着这个轴投影，每一根投影线上最大值的索引。

二维情况分析

a = np.array([[2,5,6],[7,6,1]])
print(np.argmax(a,axis=1))
[2 0]
  • 1
  • 2
  • 3

这次，我们给定axis=1，那么按照我们说的，就是沿着axis=1投影，看得出来，投影后我们只有两个值，因为[2,5,6]在一根投影线上，相当于就是我理解的一个投影线，因为这些值投影后重叠了，相当于只有一个值，所以[7,6,1]也是一个投影线，然后在这两个投影线取最大值的索引，对于[2,5,6]来说，最大值索引就是2，[7,6,1]最大值索引就是0，所以结果为[2,0] 
然后，我们分析，当我们设置axis=0的时候，也就是沿着axis=0，投影，那么[2,7]，[5,6],[6,1]分别在一个投影线上，所以结果为[1,1,0],经过验证确实是这样的。

三维情况分析

b = np.random.randint(20,size=[3,2,2])
print(b)
[[[ 0 16]
  [14  5]]

 [[16  6]
  [19  2]]

 [[11 11]
  [ 5  7]]]
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10

还是用上面的array来说，看一下结果：

print(np.argmax(b,axis=1))
[[1 0]
 [1 0]
 [0 0]]
  • 1
  • 2
  • 3
  • 4

同样的分析，我们指定了轴为1，那么也就是说我们沿着axis=1轴投影，这是一个体，那么我们投影线相当于有3*2条，其实在同一根投影线上是axis=0和axis=2的一个组合，比如axis=1中第一个上的和axis=2的第一个上的组合，也就是b[0,:,0],b[0,:,1],b[2,:,0],b[2,:,1],b[3,:,0],b[3,:,1]这六条线的索引对应的最大值，我们看他们分别为[ 0 14],[16 5],[16 19],[6 2],[11 5],[11 7],那么结果就是[[1,0],[1,0],[0,0]]。 
所以其实和二维是一样的，所谓投影线，其实可以这样表示，我们去axis=0，那么我们遍历其他维度的大小，比如 arr[:,0,0],arr[:,0,1],也就是沿着那个轴，那么轴取全部值，然后再在投影线上找最大值对应索引。

总结

tensorflow里面的是一样的用法，同时argmin()用法相同，只是他是求最小值的索引。