tf.argmax()和np.argmax()使用：

tf.argmax()和np.argmax()使用：

tf.argmax是tensorflow用numpy的np.argmax实现的，用于返回矩阵或者向量中的最大元素所在的索引值，两者使用方法相同。

import numpy as np
x=np.array([[1,2,5],[2,1,1],[5,8,2]])
y=np.argmax(x,0)#参数0表示列，参数1表示行
print(x)
print('索引值:',y+1)

运行结果如下：

[[1 2 5]
 [2 1 1]
 [5 8 2]]
索引值: [3 3 1]

注：输出索引值+1是为了方便理解最大元素所在位置，y的返回值为[2,2,0]