有关TensorFlow里的argmax()函数理解

tf.argmax()此函数是对矩阵按行或列计算最大值,并且返回这个最大值所在的下标

For example:

import numpy as np
test=np.array([[3,2,1],[4,5,6],[9,8,7]])
np.argmax(test,0)

此时 ，结果返回

array([2, 2, 2], dtype=int64)

argmax(pre,0)是指 按照数组的每一列进行比较，[3，4，5 ],[2, 5 , 6] ,[1, 6 , 7]进行比较 得出最大数所在的下标

import numpy as np
test=np.array([[3,2,1],[4,5,6],[9,8,7]])
np.argmax(test,1)

此时 ，结果返回

array([0, 2, 0], dtype=int64)

argmax(pre,1)是指 按照数组的每一行进行比较 得出最大数所在的下标